Update app.py

app.py

@@ -1,6 +1,6 @@
 from flask import Flask, request, jsonify
 import torch
-from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
+from transformers import BertTokenizer, BertForSequenceClassification
 import os
 
 app = Flask(__name__)
@@ -14,12 +14,12 @@ def load_model():
     global global_tokenizer, global_model
     try:
         print("Loading model and tokenizer...")
-        # Replace this path with your model's directory if using a custom model
-        MODEL_NAME = "distilbert-base-uncased-finetuned-sst-2-english"  # Example model, replace with your own if needed
+        # Use a different model (bert-base-uncased)
+        MODEL_NAME = "bert-base-uncased"  # You can replace this with your own fine-tuned model
         
         # Load tokenizer and model from Hugging Face Hub or a local path
-        global_tokenizer = DistilBertTokenizer.from_pretrained(MODEL_NAME)
-        global_model = DistilBertForSequenceClassification.from_pretrained(MODEL_NAME)
+        global_tokenizer = BertTokenizer.from_pretrained(MODEL_NAME)
+        global_model = BertForSequenceClassification.from_pretrained(MODEL_NAME)
         global_model.eval()
         
         print("Model loaded successfully!")