Upload 2 files

app.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import time
 import logging
 import traceback  # For better error reporting
+from PIL import Image
 
 #os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 #logger = logging.getLogger()
@@ -46,7 +47,7 @@ def load_fn(models):
     for model in models:
         if model not in models_load.keys():
             try:
-                m = gr.load(f'models/{model}')
+                m = gr.load(f'models/{model}', examples=None)
                 
                 models_load.update({model: m})
                 models_load[model] = m  # Store in dictionary
@@ -121,8 +122,8 @@ def gen_fn(model_str, prompt):
         # Execute the model's processing with a timeout
         #future = executor.submit(models_load[model_str], f"{prompt}")
         #response = future.result(timeout=150)  # Wait for result with timeout
-        response = models_load.get(model_str, lambda txt: None)(f'{prompt}')
-        if isinstance(response, gr.Image):
+        response = models_load.get(model_str, lambda txt: None).fn(f'{prompt}')
+        if isinstance(response, gr.Image) or isinstance(response, Image.Image):
             return response
         elif isinstance(response, tuple):
             return None
@@ -219,9 +220,9 @@ textarea{     position: absolute;    font-size: 1em !important;    padding: 4px;
 
 
 demo.queue()
-demo.queue = False
-demo.config["queue"] = False
-demo.launch(max_threads=40)
+#demo.queue = False
+#demo.config["queue"] = False
+demo.launch(max_threads=40, ssr_mode=False)
 
 
 

requirements.txt

@@ -0,0 +1,2 @@
+pydantic==2.10.6
+huggingface_hub==0.25.2