Update app.py
Browse files
app.py
CHANGED
|
@@ -184,7 +184,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False, document=
|
|
| 184 |
return response, None
|
| 185 |
|
| 186 |
# Main interface function
|
| 187 |
-
@spaces.GPU(duration=
|
| 188 |
def main_interface(user_prompt, image=None, audio=None, voice_only=False, websearch=False, document=None):
|
| 189 |
print("Starting main_interface function")
|
| 190 |
vqa_model.to(device='cuda', dtype=torch.bfloat16)
|
|
|
|
| 184 |
return response, None
|
| 185 |
|
| 186 |
# Main interface function
|
| 187 |
+
@spaces.GPU(duration=120)
|
| 188 |
def main_interface(user_prompt, image=None, audio=None, voice_only=False, websearch=False, document=None):
|
| 189 |
print("Starting main_interface function")
|
| 190 |
vqa_model.to(device='cuda', dtype=torch.bfloat16)
|