Upload app.py

app.py

@@ -33,7 +33,10 @@ def generate_text(prompt="Who is the CEO of Apple?"):
         stop=["#"],
     )
     output_text = output["choices"][0]["text"].strip()
-    return output_text
+
+    # Remove Prompt Echo from Generated Text
+    cleaned_output_text = output_text.replace(prompt, "")
+    return cleaned_output_text
 
 
 description = "Vicuna-7B"