Update app.py

app.py

@@ -14,22 +14,13 @@ from PIL import Image
 
 app = Flask(__name__)
 
-def load_fn(models):
-    global models_load
-    models_load = {}
-    
-    for model in models:
-        if model not in models_load.keys():
-            try:
-                m = gr.Interface.load(f'models/{model}')
-                models_load[model] = m
-                print(f"Loaded model: {model}")
-            except Exception as error:
-                print(f"Error loading model {model}: {error}")
-                m = gr.Interface(lambda txt: None, ['text'], ['image'])
-                models_load[model] = m
-
-load_fn(models)
+def load_model(model_name):
+    try:
+        # Load model within Gradio context
+        return gr.Interface.load(f'models/{model_name}')
+    except Exception as error:
+        print(f"Error loading model {model_name}: {error}")
+        return gr.Interface(lambda txt: None, ['text'], ['image'])
 
 num_models = 6
 default_models = models[:num_models]
@@ -46,10 +37,12 @@ def gen_fn(model_str, prompt, negative_prompt=None):
         return None
     noise = str('')  # Optional: str(randint(0, 99999999999))
     try:
+        # Load model within Gradio context
+        model = load_model(model_str)
         full_prompt = f'{prompt} {noise}'
         if negative_prompt:
             full_prompt += f' -{negative_prompt}'
-        result = models_load[model_str](full_prompt)
+        result = model(full_prompt)
         # Check if result is an image or a file path
         if isinstance(result, str):  # Assuming result might be a file path
             if os.path.exists(result):
@@ -72,7 +65,7 @@ def home():
     model = request.args.get('model', default_models[0] if default_models else 'NA')
     negative_prompt = request.args.get('Nprompt', None)
     
-    if not model or model not in models_load:
+    if not model:
         return f'Invalid model: {model}', 400
     
     if prompt: