KingNish commited on
Commit
5342861
1 Parent(s): e009eca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -50,12 +50,12 @@ def classify_task(prompt):
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
 
 
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