Update app.py

app.py

@@ -21,33 +21,12 @@ def load_fn(models):
     models_load = {}
     for model in models:
         if model not in models_load.keys():
-          # try:
-          #      m = gr.load(f'models/{model}')
-          #  except Exception as error:
-          #      print(f"Error loading model {model}: {error}")
-          #      m = gr.Interface(lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), enable_queue=False)
-          #  models_load.update({model: m})
             try:
-        # Attempt to generate the image
-                image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
-        
-        # Check if the image_response is a tuple, handle accordingly
-                if isinstance(image_response, tuple):
-            # If the response is a tuple, return the first item assuming it's the image
-                    image_response = image_response[0]
-        
-        # Ensure the response is an image or image-like object
-                if isinstance(image_response, gr.Image):
-                    return image_response
-                elif isinstance(image_response, str):  # If the response is a path or URL, pass it as a string
-                    return gr.Image(image_response)  # You can handle it based on your model's return type
-                else:
-                    print(f"Unexpected response type: {type(image_response)}")
-                    return None
-            
-            except Exception as e:
-                print(f"Error occurred: {e}")
-                return None
+                m = gr.load(f'models/{model}')
+            except Exception as error:
+                print(f"Error loading model {model}: {error}")
+                m = gr.Interface(lambda _: None, inputs=gr.Textbox(), outputs=gr.Image(), enable_queue=False)
+            models_load.update({model: m})
 
 
 load_fn(models)
@@ -81,16 +60,37 @@ def gen_fn(model_str, prompt, negative_prompt):
     combined_prompt = f'{prompt} {model_str} {negative_prompt} {noise}'
     print(f"Generating with prompt: {combined_prompt}")
     
-    try:
-        future = executor.submit(models_load[model_str], f'{prompt} {negative_prompt} {noise}')
-        result = future.result()
-        return result
-    except requests.exceptions.Timeout:
-        print(f"Timeout occurred for model {model_str}. Please try again later.")
-        return None
-    except Exception as e:
-        print(f"Error occurred: {e}")
-        return None
+#    try:
+#        future = executor.submit(models_load[model_str], f'{prompt} {negative_prompt} {noise}')
+#        result = future.result()
+#        return result
+#    except requests.exceptions.Timeout:
+#        print(f"Timeout occurred for model {model_str}. Please try again later.")
+#        return None
+#    except Exception as e:
+#        print(f"Error occurred: {e}")
+#        return None
+            try:
+        # Attempt to generate the image
+                image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
+        
+        # Check if the image_response is a tuple, handle accordingly
+                if isinstance(image_response, tuple):
+            # If the response is a tuple, return the first item assuming it's the image
+                    image_response = image_response[0]
+        
+        # Ensure the response is an image or image-like object
+                if isinstance(image_response, gr.Image):
+                    return image_response
+                elif isinstance(image_response, str):  # If the response is a path or URL, pass it as a string
+                    return gr.Image(image_response)  # You can handle it based on your model's return type
+                else:
+                    print(f"Unexpected response type: {type(image_response)}")
+                    return None
+            
+            except Exception as e:
+                print(f"Error occurred: {e}")
+                return None