Update app.py
Browse files
app.py
CHANGED
|
@@ -59,6 +59,7 @@ def chain():
|
|
| 59 |
memory.load_memory_variables({}) #Initialize memory
|
| 60 |
return llm_chain
|
| 61 |
|
|
|
|
| 62 |
def chat_output(inputs):
|
| 63 |
llm_chaim = chain()
|
| 64 |
result = llm_chaim.predict(input = inputs)
|
|
|
|
| 59 |
memory.load_memory_variables({}) #Initialize memory
|
| 60 |
return llm_chain
|
| 61 |
|
| 62 |
+
@spaces.GPU
|
| 63 |
def chat_output(inputs):
|
| 64 |
llm_chaim = chain()
|
| 65 |
result = llm_chaim.predict(input = inputs)
|