Update app.py

app.py

@@ -1,17 +1,30 @@
 import gradio as gr
-from comet import download_model, load_from_checkpoint
+import sys
+import torch
 
-def load_cometh_model():
-    model_path = download_model("wasanx/ComeTH")
-    return load_from_checkpoint(model_path)
+try:
+    from unbabel.comet import download_model, load_from_checkpoint
+except ImportError:
+    print("Error: unbabel-comet package not installed")
+    print("Install with: pip install unbabel-comet torch gradio")
+    sys.exit(1)
 
 def evaluate_translation(src_text, mt_text):
+    if not hasattr(evaluate_translation, "model"):
+        try:
+            model_path = download_model("wasanx/ComeTH")
+            evaluate_translation.model = load_from_checkpoint(model_path)
+        except Exception as e:
+            return f"Error loading model: {str(e)}"
+    
     translations = [{"src": src_text, "mt": mt_text}]
-    results = model.predict(translations, batch_size=1, gpus=1)
+    results = evaluate_translation.model.predict(
+        translations, 
+        batch_size=1, 
+        gpus=0 if not torch.cuda.is_available() else 1
+    )
     return float(results['scores'][0])
 
-model = load_cometh_model()
-
 demo = gr.Interface(
     fn=evaluate_translation,
     inputs=[