Update app.py

app.py

@@ -2,12 +2,11 @@ import torch
 from transformers import BertTokenizer, BertForSequenceClassification
 import gradio as gr
 
-# Load fine-tuned model
-model = BertForSequenceClassification.from_pretrained("bert-expense-classifier")
-tokenizer = BertTokenizer.from_pretrained("bert-expense-classifier")
+# Load model from local folder
+model = BertForSequenceClassification.from_pretrained(".", trust_remote_code=True)
+tokenizer = BertTokenizer.from_pretrained(".")
 
 model.eval()
-
 label_map = {0: "statement", 1: "question"}
 
 def classify_sentence(text):
@@ -17,7 +16,6 @@ def classify_sentence(text):
         predicted_class = torch.argmax(outputs.logits, dim=1).item()
     return label_map[predicted_class]
 
-# Gradio Interface
 interface = gr.Interface(
     fn=classify_sentence,
     inputs=gr.Textbox(lines=2, placeholder="Enter a sentence..."),