Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ def text_to_image(prompt):
|
|
| 6 |
img = client.predict(prompt, api_name="/predict")
|
| 7 |
return img
|
| 8 |
|
|
|
|
| 9 |
def generate(prompt):
|
| 10 |
return text_to_image(prompt)
|
| 11 |
|
|
|
|
| 6 |
img = client.predict(prompt, api_name="/predict")
|
| 7 |
return img
|
| 8 |
|
| 9 |
+
@spaces.GPU
|
| 10 |
def generate(prompt):
|
| 11 |
return text_to_image(prompt)
|
| 12 |
|