Update app.py

app.py

@@ -44,14 +44,14 @@ def load_fn(models):
         if model not in models_load.keys():
             try:
                 m = gr.load(f'models/{model}')
-                print(f"{m}\n");
+                ##print(f"{m}\n");
                 models_load.update({model: m})
                 models_load[model] = m  # Store in dictionary
             except Exception as error:
-                print(f"Error loading model {model}: {error}\n")
+                ##print(f"Error loading model {model}: {error}\n")
                 #m = gr.Interface(lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), queue=False)
                 #models_load.update({model: m})
-                traceback.print_exc()  # Prints full error stack trace for debugging
+                ##traceback.print_exc()  # Prints full error stack trace for debugging
                 #m = gr.Interface(fn=lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), queue=False)
                 models_load[model] = None
         
@@ -97,41 +97,41 @@ def gen_fn(model_str, prompt):
             return None
         elif isinstance(response, str): 
             if processed_models_count == 0:
-                print(f"***a***********")
+                ##print(f"***a***********")
               #  print(f"{prompt}")
-                print(f"{prompt}")
+                ##print(f"{prompt}")
               #  print(f"{prompt}")
-                print(f"***b***********")
+                ##print(f"***b***********")
             model_scores[current_model_name] += 1
-            print(f"OOO n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
+            ##print(f"OOO n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
             processed_models_count += 1
             if processed_models_count == len(models):
-                print("\nCycle Complete! Updated Scores:")
-                print(model_scores)
+               ## print("\nCycle Complete! Updated Scores:")
+               ## print(model_scores)
                 processed_models_count = 0
             return response
 
     except TimeoutError:
-        print(f"TimeoutError: Model '{model_str}' did not respond within 150 seconds.")
+        ##print(f"TimeoutError: Model '{model_str}' did not respond within 150 seconds.")
         processed_models_count += 1
         if processed_models_count == len(models):
-            print("\nCycle Complete! Updated Scores:")
-            print(model_scores)
+          ##  print("\nCycle Complete! Updated Scores:")
+          ##  print(model_scores)
             processed_models_count = 0
         return None
 
     except Exception as e:
         if processed_models_count == 0:
-            print(f"******c*******")
+         ##   print(f"******c*******")
           #  print(f"{prompt}")
           #  print(f"{prompt}")
           #  print(f"{prompt}")
-            print(f"******d*******")
-        print(f"--- n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
+           ## print(f"******d*******")
+        ##print(f"--- n:{processed_models_count} x:{current_model_index}  r[{model_scores[current_model_name]}] {model_str}")
         processed_models_count += 1
         if processed_models_count == len(models):
-            print("\nCycle Complete! Updated Scores:")
-            print(model_scores)
+          ##  print("\nCycle Complete! Updated Scores:")
+          ##  print(model_scores)
             processed_models_count = 0
         return None