KingNish commited on
Commit
e009eca
1 Parent(s): 599427e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -50,13 +50,12 @@ def classify_task(prompt):
50
  for response in stream:
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():
57
- return 'Image Generation'
58
- else:
59
- return 'Unknown Task'
60
 
61
 
62
 
 
50
  for response in stream:
51
  if not response.token.text == "</s>":
52
  output += response.token.text
53
+ if 'text' in output.lower():
54
+ user = text(prompt)
55
+ elif 'image' in output.lower():
56
+ return 'Image Generation'
57
+ else:
58
+ return 'Unknown Task'
 
59
 
60
 
61