Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,8 +51,6 @@ def classify_task(prompt):
|
|
| 51 |
if not response.token.text == "</s>":
|
| 52 |
output += response.token.text
|
| 53 |
|
| 54 |
-
yield output
|
| 55 |
-
|
| 56 |
if 'text' in output.lower():
|
| 57 |
user = text(prompt)
|
| 58 |
elif 'image' in output.lower():
|
|
|
|
| 51 |
if not response.token.text == "</s>":
|
| 52 |
output += response.token.text
|
| 53 |
|
|
|
|
|
|
|
| 54 |
if 'text' in output.lower():
|
| 55 |
user = text(prompt)
|
| 56 |
elif 'image' in output.lower():
|