Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -135,7 +135,7 @@ class ModelWrapper:
|
|
135 |
)
|
136 |
return prompt
|
137 |
|
138 |
-
@spaces.GPU(duration=
|
139 |
def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
|
140 |
enable_thinking=True, system_prompt=SYSTEM_PROMPT):
|
141 |
print("Generating response...")
|
|
|
135 |
)
|
136 |
return prompt
|
137 |
|
138 |
+
@spaces.GPU(duration=120)
|
139 |
def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
|
140 |
enable_thinking=True, system_prompt=SYSTEM_PROMPT):
|
141 |
print("Generating response...")
|