Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
 import fitz  # PyMuPDF
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-
+# Load model directly
+model_name = "openai-community/gpt2"
 # Load the LLaMA model and tokenizer
-model_name = "meta-llama/Meta-Llama-Guard-2-8B"  # Update this with the correct LLaMA model name
+#model_name = "meta-llama/Meta-Llama-Guard-2-8B"  # Update this with the correct LLaMA model name
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)