Update app.py

app.py

@@ -1,7 +1,7 @@
 import fitz  # PyMuPDF
 import re
 from datasets import Dataset
-from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer
+from transformers import AutoModelForCausalLM, TrainingArguments, Trainer
 import gradio as gr
 from transformers import pipeline
 
@@ -41,7 +41,7 @@ def tokenize_function(examples):
 
 tokenized_datasets = dataset.map(tokenize_function, batched=True)
 
-model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)  # Adjust for task
+model = AutoModelForCausalLM.from_pretrained(model_name)  # Adjust for task
 
 training_args = TrainingArguments(
     output_dir="./results",