Update app.py

app.py

@@ -43,30 +43,6 @@ extend_prompt_1 = True
 extend_prompt_2 = True
 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>]"
-if extend_prompt_2:
-    def load_prompter():
-      prompter_model = AutoModelForCausalLM.from_pretrained("microsoft/Promptist")
-      tokenizer = AutoTokenizer.from_pretrained("gpt2")
-      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 
-        outputs = prompter_model.generate(input_ids, do_sample=False, max_new_tokens=75, num_beams=8, num_return_sequences=8, eos_token_id=eos_id, pad_token_id=eos_id, length_penalty=-1.0)
-        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>]"
-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>]"
-
 do_dreamlike_photoreal = False
 if do_dreamlike_photoreal:
     def add_random_noise(prompt, noise_level=0.1):
@@ -88,7 +64,6 @@ if do_dreamlike_photoreal:
 
     tab_actions.append(dreamlike_image)
     tab_titles.append("Dreamlike_2.0")
-    thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0' _blank><font style='color:blue;weight:bold;'>dreamlike-photoreal-2.0</font></a>]"
 
 for space_id in space_ids.keys():
     print(space_id, space_ids[space_id])
@@ -101,7 +76,6 @@ for space_id in space_ids.keys():
         else:
             tab_content = gr.Interface.load(space_id)
             tab_actions.append(tab_content)
-            thanks_info += f"[<a style='display:inline-block' href='https://huggingface.co/{space_id}' _blank><font style='color:blue;weight:bold;'>{tab_title}</font></a>]"
     except Exception as e:
         logger.info(f"load_fail__{space_id}_{e}")
 
@@ -333,7 +307,7 @@ def prompt_draw_2(prompt):
         prompt_zh = getTextTrans(prompt, source='en', target='zh')
     return prompt_en, prompt_zh
 
-with gr.Blocks(title='Text-to-Image') as demo:
+with gr.Blocks(title='VM8 Text2Image') as demo:
     with gr.Group(elem_id="page_1", visible=True) as page_1:
         with gr.Box():            
             with gr.Row():
@@ -374,8 +348,6 @@ with gr.Blocks(title='Text-to-Image') as demo:
             with gr.Row():                
                 switch_tab_advanced_btn = gr.Button(value = "Switch_tab_advanced", elem_id="switch_tab_advanced_btn")
                 switch_tab_advanced_btn.click(fn=None, inputs=[], outputs=[], _js=switch_tab_advanced)   
-            with gr.Row():                
-                gr.HTML(f"<p>{thanks_info}</p>") 
 
             if extend_prompt_1:
                 extend_btn_1.click(fn=prompt_extend_1, inputs=[prompt_input0], outputs=[prompt_input1])