ysharma HF staff commited on
Commit
56d5f94
1 Parent(s): 02a5461
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -14,13 +14,14 @@ import os
14
  #API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
15
  HF_TOKEN = os.environ["HF_TOKEN"]
16
  #headers = {"Authorization": f"Bearer {HF_TOKEN}"}
17
-
 
18
  def get_sd(translated_txt):
19
  print("******** Inside get_SD ********")
20
  print(f"translated_txt is : {translated_txt}")
21
- sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token='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')
22
  print(f"stable Diff inf is : {sd_inf}")
23
- tbox, sd_img_gallery = sd_infa(translated_txt, '4', '45', '7.5', 1024, fn_index=1) #(prompt, samples, steps, scale, seed)
24
 
25
  return sd_img_gallery[0]
26
 
 
14
  #API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
15
  HF_TOKEN = os.environ["HF_TOKEN"]
16
  #headers = {"Authorization": f"Bearer {HF_TOKEN}"}
17
+ sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token='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')
18
+
19
  def get_sd(translated_txt):
20
  print("******** Inside get_SD ********")
21
  print(f"translated_txt is : {translated_txt}")
22
+ #sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token='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')
23
  print(f"stable Diff inf is : {sd_inf}")
24
+ tbox, sd_img_gallery = sd_inf(translated_txt, '4', '45', '7.5', 1024, fn_index=1) #(prompt, samples, steps, scale, seed)
25
 
26
  return sd_img_gallery[0]
27