Update app.py

app.py

@@ -50,13 +50,12 @@ def classify_task(prompt):
     for response in stream:
         if not response.token.text == "</s>":
             output += response.token.text
-
-    if 'text' in output.lower():
-        user = text(prompt)
-    elif 'image' in output.lower():
-        return 'Image Generation'
-    else:
-        return 'Unknown Task'
+                if 'text' in output.lower():
+                    user = text(prompt)
+                elif 'image' in output.lower():
+                    return 'Image Generation'
+                else:
+                    return 'Unknown Task'