Update app.py
Browse files
app.py
CHANGED
|
@@ -393,7 +393,7 @@ def create_ui():
|
|
| 393 |
return demo
|
| 394 |
|
| 395 |
# Main interface function
|
| 396 |
-
@spaces.GPU()
|
| 397 |
def main_interface(user_prompt, image=None, audio=None, voice_only=False, websearch=False, document=None):
|
| 398 |
print("Starting main_interface function")
|
| 399 |
vqa_model.to(device='cuda', dtype=torch.bfloat16)
|
|
|
|
| 393 |
return demo
|
| 394 |
|
| 395 |
# Main interface function
|
| 396 |
+
@spaces.GPU(duration=180)
|
| 397 |
def main_interface(user_prompt, image=None, audio=None, voice_only=False, websearch=False, document=None):
|
| 398 |
print("Starting main_interface function")
|
| 399 |
vqa_model.to(device='cuda', dtype=torch.bfloat16)
|