Update app.py

app.py

@@ -40,32 +40,23 @@ def add_new_eval(
     family_prediction_dataset,
     save,
 ):
-    gr.Info.update(
-        value="Your submission is being processed...",
-        show_progress="full",
-        visible=True,
-        elem_id="processing-modal"
-    )
+    processing_info = gr.Info("Your submission is being processed...", elem_id="processing-modal")
 
     representation_name = model_name_textbox if revision_name_textbox == '' else revision_name_textbox
-    results = run_probe(benchmark_types, representation_name, human_file, skempi_file, similarity_tasks, function_prediction_aspect, function_prediction_dataset, family_prediction_dataset)
-    
 
+    try:
+        results = run_probe(benchmark_types, representation_name, human_file, skempi_file, similarity_tasks, function_prediction_aspect, function_prediction_dataset, family_prediction_dataset)
+    except:
+        completion_info = gr.Info("Your submission has not been processed. Please check your representation files!", elem_id="completion-modal")
+
+    
     if save:
         save_results(representation_name, benchmark_types, results)
-        gr.Info.update(
-            value="Your submission has been processed and results are saved!",
-            show_progress="full",
-            visible=True,
-            elem_id="completion-modal"
-        )
+        completion_info = gr.Info("Your submission has been processed and results are saved!", elem_id="completion-modal")
+
     else:
-        gr.Info.update(
-            value="Your submission has been processed!",
-            show_progress="full",
-            visible=True,
-            elem_id="completion-modal"
-        )
+        completion_info = gr.Info("Your submission has been processed!", elem_id="completion-modal")
+
     return 0
 
 def refresh_data():