Update app.py

app.py

@@ -11,7 +11,7 @@ print("B")
 #os.system('pip install tokenizers==0.8.1rc1')
 print("C")
 
-os.system('wget https://huggingface.co/yxchng/elia_refcoco/resolve/main/model_best_refcoco_0508.pth')
+os.system('wget https://huggingface.co/yxchng/magnet_weights/resolve/main/magnet_refcoco.pth')
 image_path = './image001.png'
 sentence = 'spoon on the dish'
 weights = 'model_best_refcoco_0508.pth'
@@ -45,6 +45,7 @@ from addict import Dict
 #from bert.modeling_bert import BertLMPredictionHead, BertEncoder
 import cv2
 import textwrap
+import spaces
 
 class WrapperModel(nn.Module):
     def __init__(self, image_model, language_model, classifier) :
@@ -252,7 +253,7 @@ def overlay_davis(image, mask, colors=[[0, 0, 0], [255, 0, 0]], cscale=1, alpha=
 
     return im_overlay.astype(image.dtype)
 
-
+@spaces.GPU(duration=10)
 def run_model(img, sentence):
 
 #img = Image.open(image_path).convert("RGB")