Update app.py

app.py

@@ -7,7 +7,7 @@ model_name = "unsloth/Llama-3.2-1B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     device_map="auto"
 )
 
@@ -68,10 +68,7 @@ demo = gr.ChatInterface(
         "Write a short poem about coding",
         "Explain quantum computing in simple terms"
     ],
-    theme=gr.themes.Soft(),
-    retry_btn="Retry",
-    undo_btn="Undo",
-    clear_btn="Clear"
+    theme=gr.themes.Soft()
 )
 
 if __name__ == "__main__":