Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
 from transformers import pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 def generateYodaLanguage(prompt):
-  generator = pipeline("text-generation", model="DrElaheJ/phi3-mini-yoda-adapter")
+  repo_id="DrElaheJ/phi3-mini-yoda-adapter"
+  model = AutoModelForCausalLM.from_pretrained (repo_id)
+  tokenizer = AutoTokenizer.from_pretrained(repo_id)
+  generator = pipeline("text-generation", model=repo_id)
   output = generator([{"role": "user", "content": prompt}], max_new_tokens=128, return_full_text=False)[0]
   return output["generated_text"]