update for ernie

app.py

@@ -8,8 +8,9 @@ import paddlehub as hub
 HF_TOKEN = os.environ["HF_TOKEN"]
 model = hub.Module(name='ernie_vilg')
 
-def get_ernie_vilg(text_prompts): 
-  results = model.generate_image(text_prompts=text_prompts, style="油画", visualization=False)
+def get_ernie_vilg(text_prompts, style): 
+  style = style.split('-')[0]
+  results = model.generate_image(text_prompts=text_prompts, style=style, visualization=False)
   return results[0]
   
 sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token=HF_TOKEN)
@@ -30,7 +31,7 @@ def get_chinese_translation(in_language_first, in_language_second, text):
   
   return output 
   
-  
+#Block inference not working for stable diffusion   
 def get_sd(translated_txt, samples, steps, scale, seed):
   print("******** Inside get_SD ********")
   print(f"translated_txt is : {translated_txt}")
@@ -45,27 +46,29 @@ with demo:
   
   with gr.Row():
     in_text_prompt = gr.Textbox(label="Enter English text here")
-    out_text_chinese = gr.Textbox(label="Your Chinese language output")
-    b1 = gr.Button("English to Chinese")
+    out_text_chinese = gr.Textbox(label="Text in Simplified Chinese")
+    b1 = gr.Button("English to Simplified Chinese")
     
     s1 = gr.Slider(label='samples', value=4, visible=False)
     s2 = gr.Slider(label='steps', value=45, visible=False)
     s3 = gr.Slider(label='scale', value=7.5, visible=False)
     s4 = gr.Slider(label='seed', value=1024, visible=False)
     
+    
     in_language_first = gr.Dropdown(visible=False, value= 'eng_Latn') #'English') 
     in_language_second = gr.Dropdown(visible=False, value= 'zho_Hans') #'Chinese (Simplified)') 
     
   with gr.Row():
+    in_styles = gr.Dropdown(['水彩-WaterColor', '油画-OilPainting', '粉笔画-Painting', '卡通-Cartoon', '蜡笔画-Pencils', '儿童画-ChildrensPaintings', '探索无限-ExploringTheInfinite']),
     out_ernie = gr.Image(type="pil", label="Ernie output for the given prompt")
     #out_gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery") #.style(grid=[2, 3], height="auto")
     b2 = gr.Button("Generate Images from Ernie")
     
-    out_sd = gr.Image(type="pil", label="SD output for the given prompt")
-    b3 = gr.Button("Generate Images from SD")
+    #out_sd = gr.Image(type="pil", label="SD output for the given prompt")
+    #b3 = gr.Button("Generate Images from SD")
     
     b1.click(get_chinese_translation, [in_language_first, in_language_second, in_text_prompt], out_text_chinese )  
-    b2.click(get_ernie_vilg, out_text_chinese, out_ernie) #out_gallery )  
-    b3.click(get_sd, [in_text_prompt,s1,s2,s3,s4], out_sd) #out_gallery )  
+    b2.click(get_ernie_vilg, [out_text_chinese, in_styles], out_ernie) #out_gallery )  
+    #b3.click(get_sd, [in_text_prompt,s1,s2,s3,s4], out_sd) #out_gallery )  
     
 demo.launch(enable_queue=True, debug=True)