Update app.py

app.py

@@ -3,11 +3,12 @@ import gradio as gr
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf") #meta-llama/Llama-2-7b-hf 
-model = AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
+model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
 
 def eval_text(text):
     # Encode the input text
+    text = "Eres un experto en lenguaje claro. Evalúa el texto siguiente y di si es muy claro, claro o poco claro. El texto es este: " + text
     input_ids = tokenizer.encode(text, return_tensors="pt")
 
 # Generate text
@@ -30,6 +31,6 @@ def eval_text(text):
     return(f"Result: {generation[0]['generated_text']}")
 
 
-demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="Llama2")
+demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="microsoft/phi-2")
     
 demo.launch(share=True)