Update app.py

app.py

@@ -9,22 +9,16 @@ from t2v_metrics import VQAScore, list_all_vqascore_models
 print(list_all_vqascore_models())
 
 # Initialize the model only once
-model_pipe = None
-
-@spaces.GPU
-def initialize_model(model_name):
-    global model_pipe
-    if model_pipe is None:
-        model_pipe = VQAScore(model=model_name)  # our recommended scoring model
-        print("Model initialized!")
-    return model_pipe
+if torch.cuda.is_available():
+    model_pipe = VQAScore(model="clip-flant5-x")  # our recommended scoring model
+    model_pipe.to("cuda")
+    print("Model initialized!")
 
 @spaces.GPU
 def generate(model_name, image, text):
-    print("Model_name:", model_name)
+    # print("Model_name:", model_name)
     print("Image:", image)
     print("Text:", text)
-    model_pipe = initialize_model(model_name)
     return model_pipe(images=[image], texts=[text])
 
 iface = gr.Interface(