Update app.py

app.py

@@ -2,9 +2,12 @@
 
 #!/usr/bin/env python
 # encoding: utf-8
+import spaces
 import torch
-torch.randn(10).cuda()
-
+@spaces.GPU
+def debug():
+    torch.randn(10).cuda()
+debug()
 import argparse
 from transformers import AutoModel, AutoTokenizer
 import gradio as gr
@@ -110,7 +113,7 @@ def create_multimodal_input(upload_image_disabled=False, upload_video_disabled=F
                                         upload_video_button_props={'label': 'Upload Video', 'disabled': upload_video_disabled, 'file_count': 'single'},
                                         submit_button_props={'label': 'Submit'})
 
-
+@spaces.GPU
 def chat(img, msgs, ctx, params=None, vision_hidden_states=None):
     try:
         print('msgs:', msgs)