Spaces:
Running
Running
lalashechka
commited on
Commit
•
0bbffeb
1
Parent(s):
9147c08
Update app.py
Browse files
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("
|
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 |
|