Update app.py

app.py

@@ -6,10 +6,11 @@ import io
 from torch.utils.data import Dataset
 from sklearn.metrics import classification_report
 
-# Load the tokenizer and model
-model_path = "model"
-tokenizer = GPT2Tokenizer.from_pretrained(model_path, local_files_only=True)
-model = LlamaForSequenceClassification.from_pretrained(model_path, local_files_only=True)
+# Load Model and Tokenizer
+tokenizer = GPT2Tokenizer.from_pretrained("jatinmehra/smolLM-fined-tuned-for-PLAGAIRISM_Detection")
+
+model = LlamaForSequenceClassification.from_pretrained("jatinmehra/smolLM-fined-tuned-for-PLAGAIRISM_Detection", num_labels=2)
+
 model.eval()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")