Update app.py

app.py

@@ -11,7 +11,6 @@ print(list_all_vqascore_models())
 # Initialize the model only once
 if torch.cuda.is_available():
     model_pipe = VQAScore(model="clip-flant5-xl")  # our recommended scoring model
-    model_pipe.to("cuda")
     print("Model initialized!")
 
 @spaces.GPU
@@ -19,6 +18,7 @@ def generate(model_name, image, text):
     # print("Model_name:", model_name)
     print("Image:", image)
     print("Text:", text)
+    model_pipe.to("cuda")
     return model_pipe(images=[image], texts=[text])
 
 iface = gr.Interface(