Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 
-# Load your quantized model
+# Load your model
 model_name = "khaled123/hf"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 def chat(input_text):
     inputs = tokenizer(input_text, return_tensors="pt")
@@ -17,6 +16,7 @@ iface = gr.Interface(fn=chat,
                      inputs="text", 
                      outputs="text",
                      title="Chatbot",
-                     description="A chatbot using a quantized model")
+                     description="A chatbot using a model")
 
 iface.launch()
+