Update app.py

app.py

@@ -223,6 +223,8 @@ def call_main(purpose, task, history, action_input):
             continue
         if line.startswith("thought: "):
             history += "{}\n".format(line)
+        elif line.startswith("action: COMPLETE"):
+            return "COMPLETE", None, history, task
         elif line.startswith("action: "):
             action_name, action_input = parse_action(line)
             history += "{}\n".format(line)
@@ -308,9 +310,9 @@ def run(purpose,history):
         if action_name == "COMPLETE":
             return history
 examples =[
-    "find a model that I can use to generate an image by providing a text prompt",
+    "find the most popular model that I can use to generate an image by providing a text prompt",
     "return the top 10 models that I can use to identify objects in images",
-    "which models are the most popular right now?"
+    "which models have the most likes from each category?"
 ]