Spaces:
Running
on
Zero
Running
on
Zero
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():
|