Update app.py

app.py

@@ -72,16 +72,9 @@ def gen_fn(model_str, prompt, negative_prompt):
     print(f"Generating with prompt: {combined_prompt}")
     
     try:
-        # Attempt to generate the image
-        #image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
-        print(f"77 {models_load[model_str](f'{combined_prompt}')}")
-        image_response = models_load[model_str](f'{combined_prompt}')
-        # Check if the image_response is a tuple, handle accordingly
-       # if isinstance(image_response, tuple):
-       #     print(f"{image_response}")
-       #     # If the response is a tuple, assume the first item is the image
-       #     image_response = image_response[0]
-        
+        image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
+      # print(f"77 {models_load[model_str](f'{combined_prompt}')}")
+    #   image_response = models_load[model_str](f'{combined_prompt}')
         # Ensure the response is an image or image-like object
         if isinstance(image_response, gr.Image):
             return image_response
@@ -93,8 +86,7 @@ def gen_fn(model_str, prompt, negative_prompt):
     except Exception as e:
         print(f"Error occurred: {e}")
         return None
-        #None
-
+      
 
 
 def make_me():
@@ -105,7 +97,7 @@ def make_me():
         stop_button = gr.Button('Stop', variant='secondary', interactive=False)
         gen_button.click(lambda _: gr.update(interactive=True), None, stop_button)
         
-        log_output = gr.Textbox(label="Logs", lines=10, interactive=False)
+        log_output = gr.Textbox(label="Logs", lines=2, interactive=False)
         log_button = gr.Button("Start Logging")
         log_button.click(log_message, outputs=log_output)