Update app.py

app.py

@@ -15,7 +15,8 @@ def classificate_with_gpt3(text, labels, engine="text-similarity-davinci-001"):
     model=engine,
     prompt=prompt,
     temperature=0.7,
-    max_tokens=1000,
+    max_tokens=10,
+    stop: ["\n"],
     top_p=1,
     frequency_penalty=0,
     presence_penalty=0
@@ -24,7 +25,7 @@ def classificate_with_gpt3(text, labels, engine="text-similarity-davinci-001"):
   return completion
  
 def gpt3_zero_shot_classification(text, labels):  
-  completion = classificate_with_gpt3(text, labels)
+  completion = classificate_with_gpt3(text, labels, engine="text-babbage-001")
   return completion
 
 # Gradio UI