files changed

app.py

@@ -1,6 +1,15 @@
 import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-def greet(name):
+model = 
+AutoModelForCausalLM.from_pretrained("Sahibsingh12/phi-1-5-finetuned-cazton_complete", 
+trust_remote_code=True, torch_dtype=torch.float32)
+tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", 
+trust_remote_code=True)
+
+
+ def greet(name):
     return "Hello " + name + "!!"
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")