Upload app.py

app.py

@@ -57,7 +57,10 @@ def gen_fn_gallery(model_str, prompt, gallery):
     rand = randint(1, 500)
     for i in range(rand):
         noise += " "
-    gallery.append(models_load[model_str](f'{prompt} {noise}'))
+    try:
+        gallery.append(models_load[model_str](f'{prompt} {noise}'))
+    except Exception as e:
+        print(e)
     return gallery