updated

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+tokenizer = AutoTokenizer.from_pretrained("<lehmanc25>/my-phi2-model")
+model = AutoModelForCausalLM.from_pretrained("<lehmanc25>/my-phi2-model")
 
 def chat_with_phi2(input_text):
-    # Encoding the input text
     encoded_input = tokenizer(input_text, return_tensors='pt')
-    # Generating output using the fine-tuned model
     output = model.generate(**encoded_input, max_length=50)
-    # Decoding the output tokens to a string
     response = tokenizer.decode(output[0], skip_special_tokens=True)
     return response
 
@@ -17,5 +18,4 @@ interface = gr.Interface(
     description="Ask any question about Washington and Lee University!"
 )
 
-interface.launch()