Update app.py

app.py

@@ -137,7 +137,12 @@ def gen_fn(model_str, prompt):
     if model_str == 'NA':
         return None
     noise = str(randint(0, 9999))
-    return models_load[model_str](f'{prompt} {noise}')
+    try :
+        m=models_load[model_str](f'{prompt} {noise}')
+    except Exception as error : 
+        print(error + " : " + model_str)
+    
+    return m
 
 def make_me():
    # with gr.Tab('The Dream'):