Update app.py

app.py

@@ -3,6 +3,27 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 import pytesseract
 
 
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
+model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="meta-llama/Llama-3.3-70B-Instruct")
+pipe(messages)
+
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.3-70B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.3-70B-Instruct")
+
+
 # Initialize chat model (You can change the model here)
 chat_model = pipeline("text-generation", model="gpt2")  # You can switch to any model of your choice