Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import pipeline
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import torch
 import gradio as gr
 import os
@@ -11,6 +11,9 @@ if not token:
 
  
 model_id = "meta-llama/Llama-3.2-1B-Instruct"
+tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
+model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=token)
+
 pipe = pipeline(
     "text-generation",
     model=model_id,