Update app.py

app.py

@@ -76,7 +76,6 @@ def gen_fn(model_str, prompt, negative_prompt):
         #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}')
-        print(f"{image_response}")
         # Check if the image_response is a tuple, handle accordingly
        # if isinstance(image_response, tuple):
        #     print(f"{image_response}")
@@ -86,11 +85,11 @@ def gen_fn(model_str, prompt, negative_prompt):
         # 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
+#        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 e
@@ -138,7 +137,7 @@ js_code = """
         window.scrollTo = function() {};};
     setTimeout(() => {
         window.scrollTo = originalScroll;
-    }, 300000); // Restore scroll function after 3 seconds
+    }, 1000); // Restore scroll function after 3 seconds
 </script>
 """