Update app.py

app.py

@@ -28,16 +28,21 @@ models =[
 ]
 
 
-interfaces = {}
+model_functions = {}
 model_idx = 1
 for model_path in models:
-    interfaces[model_idx] = gr.Interface.load(f"models/{model_path}", live=False, preprocess=True, postprocess=False)
+    try:
+        model_functions[model_idx] = gr.Interface.load(f"models/{model_path}", live=False, preprocess=True, postprocess=False)
+    except Exception as error:
+        def the_fn(txt):
+            return None
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image"])
     model_idx+=1
 
 
 def send_it_idx(idx):
     def send_it_fn(prompt):
-        output = (interfaces.get(str(idx)) or interfaces.get(str(1)))(prompt)
+        output = (model_functions.get(str(idx)) or model_functions.get(str(1)))(prompt)
         return output
     return send_it_fn
 
@@ -82,7 +87,7 @@ def clear_fn():
 
 
 
-with gr.Blocks() as my_interface:
+with gr.Blocks(title="SD Models") as my_interface:
     with gr.Column(scale=12):
         # with gr.Row():
         #     gr.Markdown("""- Primary prompt: 你想画的内容(英文单词，如 a cat, 加英文逗号效果更好；点 Improve 按钮进行完善)\n- Real prompt: 完善后的提示词，出现后再点右边的 Run 按钮开始运行""")
@@ -123,7 +128,7 @@ with gr.Blocks() as my_interface:
         runs_dict = {}
         model_idx = 1
         for model_path in models:
-            runs_dict[model_idx] = run.click(interfaces[model_idx], inputs=[primary_prompt], outputs=[sd_outputs[model_idx]])
+            runs_dict[model_idx] = run.click(model_functions[model_idx], inputs=[primary_prompt], outputs=[sd_outputs[model_idx]])
             model_idx += 1
             pass
         pass