ysharma HF staff commited on
Commit
d82d4ca
1 Parent(s): 0978583

update gradio version and add progress bar

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -29,12 +29,17 @@ def get_model_list(category):
29
 
30
  def get_initial_prompt(model_nm):
31
  #Example - a photo of shbrcky dog
 
 
32
  user_model_nm = model_nm.split('/')[-1]
33
  if '-' in user_model_nm:
34
  prompt = " ".join(user_model_nm.split('-'))
35
  else:
36
  prompt = user_model_nm
37
- return gr.Textbox.update(value="a photo of " + prompt + " ")
 
 
 
38
 
39
  def get_pipeline(model_name): #, progress=gr.Progress(track_tqdm=True)):
40
  #Using diffusers pipeline to generate an image for the demo
 
29
 
30
  def get_initial_prompt(model_nm):
31
  #Example - a photo of shbrcky dog
32
+ print(******Inside get initial prompt******)
33
+ print(f"model name from Gradio input is - {model_nm}")
34
  user_model_nm = model_nm.split('/')[-1]
35
  if '-' in user_model_nm:
36
  prompt = " ".join(user_model_nm.split('-'))
37
  else:
38
  prompt = user_model_nm
39
+ print(f"Extracted name for constructing Prompt is - {user_model_nm}")
40
+ print(f"Extracted name for constructing Prompt is - {prompt}")
41
+ prompt = "a photo of " + prompt + " "
42
+ return gr.Textbox.update(value=prompt)
43
 
44
  def get_pipeline(model_name): #, progress=gr.Progress(track_tqdm=True)):
45
  #Using diffusers pipeline to generate an image for the demo