Update app.py

app.py

@@ -5,13 +5,13 @@ torch.jit.script = lambda f: f  # Avoid script error in lambda
 
 from t2v_metrics import VQAScore, list_all_vqascore_models
 
+def update_model(model_name):
+    return VQAScore(model=model_name, device="cuda") 
+
 # Global model variable, but do not initialize or move to CUDA here
 cur_model_name = "clip-flant5-xl"
 model_pipe = update_model(cur_model_name)
 
-def update_model(model_name):
-    return VQAScore(model=model_name, device="cuda") 
-
 @spaces.GPU(duration = 20)
 def generate(model_name, image, text):
     if model_name != cur_model_name: