Upload app.py

app.py

@@ -23,7 +23,7 @@ download_file(ggml_model_path, filename)
 llm = Llama(model_path=filename, n_ctx=512, n_batch=126)
 
 
-def generate_text(prompt):
+def generate_text(prompt="Who is the CEO of Apple?"):
     output = llm(
         prompt,
         max_tokens=256,
@@ -32,14 +32,14 @@ def generate_text(prompt):
         echo=False,
         stop=["#"],
     )
-    output_text = output["choices"][0]["text"]
+    output_text = output["choices"][0]["text"].strip()
     return output_text
 
 
 description = "Vicuna-7B"
 
 examples = [
-    ["What is the capital of France? ", "The capital of France is Paris."],
+    ["What is the capital of France?", "The capital of France is Paris."],
     [
         "Who wrote the novel 'Pride and Prejudice'?",
         "The novel 'Pride and Prejudice' was written by Jane Austen.",