update app.y

app.py

@@ -55,8 +55,12 @@ extend_prompt_3 = True
 
 thanks_info = "Thanks: "
 if extend_prompt_1:
-    extend_prompt_pipe = pipeline('text-generation', model='yizhangliu/prompt-extend', max_length=77, pad_token_id=0)
-    thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/daspartho/prompt-extend' _blank><font style='color:blue;weight:bold;'>prompt-extend(1)</font></a>]"
+    try:
+        extend_prompt_pipe = pipeline('text-generation', model='yizhangliu/prompt-extend', max_length=77, pad_token_id=0)
+        thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/daspartho/prompt-extend' _blank><font style='color:blue;weight:bold;'>prompt-extend(1)</font></a>]"
+    except Exception as e:
+        extend_prompt_1 = False
+        
 if extend_prompt_2:
     def load_prompter():
       prompter_model = AutoModelForCausalLM.from_pretrained("microsoft/Promptist")
@@ -64,7 +68,6 @@ if extend_prompt_2:
       tokenizer.pad_token = tokenizer.eos_token
       tokenizer.padding_side = "left"
       return prompter_model, tokenizer
-    prompter_model, prompter_tokenizer = load_prompter()
     def extend_prompt_microsoft(in_text):
         input_ids = prompter_tokenizer(in_text.strip()+" Rephrase:", return_tensors="pt").input_ids
         eos_id = prompter_tokenizer.eos_token_id 
@@ -72,10 +75,18 @@ if extend_prompt_2:
         output_texts = prompter_tokenizer.batch_decode(outputs, skip_special_tokens=True)
         res = output_texts[0].replace(in_text+" Rephrase:", "").strip()
         return res    
-    thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/microsoft/Promptist' _blank><font style='color:blue;weight:bold;'>Promptist(2)</font></a>]"
+    try:
+        prompter_model, prompter_tokenizer = load_prompter()
+        thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/microsoft/Promptist' _blank><font style='color:blue;weight:bold;'>Promptist(2)</font></a>]"
+    except Exception as e:
+        extend_prompt_2 = False
+
 if extend_prompt_3:
-    MagicPrompt = gr.Interface.load("spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
-    thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/Gustavosta/MagicPrompt-Stable-Diffusion' _blank><font style='color:blue;weight:bold;'>MagicPrompt(3)</font></a>]"
+    try:
+        MagicPrompt = gr.Interface.load("spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
+        thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/spaces/Gustavosta/MagicPrompt-Stable-Diffusion' _blank><font style='color:blue;weight:bold;'>MagicPrompt(3)</font></a>]"
+    except Exception as e:
+        extend_prompt_3 = False
 
 do_dreamlike_photoreal = False
 if do_dreamlike_photoreal: