Update app.py

app.py

@@ -29,25 +29,25 @@ def load_fn(models):
           #  models_load.update({model: m})
             try:
         # Attempt to generate the image
-        image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
+                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 isinstance(image_response, tuple):
             # If the response is a tuple, return the first item assuming it's the image
-            image_response = image_response[0]
+                    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
+                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
+            except Exception as e:
+                print(f"Error occurred: {e}")
+                return None
 
 
 load_fn(models)