Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
artificialguybr
commited on
Commit
•
e24c40e
1
Parent(s):
5856490
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def run_lora(prompt):
|
|
23 |
token = os.getenv("API_TOKEN")
|
24 |
payload = {"inputs": f"{prompt} {trigger_word}"}
|
25 |
image_bytes = query(payload, api_url, token)
|
26 |
-
return Image.open(image_bytes)
|
27 |
|
28 |
# Gradio UI
|
29 |
print("Before Gradio Interface")
|
|
|
23 |
token = os.getenv("API_TOKEN")
|
24 |
payload = {"inputs": f"{prompt} {trigger_word}"}
|
25 |
image_bytes = query(payload, api_url, token)
|
26 |
+
return Image.open(image_bytes)
|
27 |
|
28 |
# Gradio UI
|
29 |
print("Before Gradio Interface")
|