lalashechka commited on
Commit
0bbffeb
1 Parent(s): 9147c08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -44,10 +44,9 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
44
  url_sd5 = os.getenv("url_sd5")
45
  url_sd6 = os.getenv("url_sd6")
46
  hf_token = os.getenv("hf_token")
47
- hf_token = str(hf_token)
48
  if task == "Playground v2":
49
  playground = str(os.getenv("playground"))
50
- client = Client("https://playgroundai-playground-v2.hf.space", hf_token=hf_token)
51
  result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
52
  return result[0][0]['image']
53
 
 
44
  url_sd5 = os.getenv("url_sd5")
45
  url_sd6 = os.getenv("url_sd6")
46
  hf_token = os.getenv("hf_token")
 
47
  if task == "Playground v2":
48
  playground = str(os.getenv("playground"))
49
+ client = Client("playgroundai/playground-v2", hf_token=hf_token)
50
  result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
51
  return result[0][0]['image']
52