Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import pipeline, AutoTokenizer
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 from langdetect import detect, DetectorFactory
 
 # Ensure consistent language detection results
@@ -16,14 +16,14 @@ app = FastAPI()
 # Load the original tokenizer from the base model
 original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
 
-# Load the fine-tuned model and pass the tokenizer explicitly
+# Load the fine-tuned model (johndoee/sentiment) and pass the tokenizer explicitly
 multilingual_model = pipeline(
     "sentiment-analysis",
-    model="./model",
+    model="johndoee/sentiment",
     tokenizer=original_tokenizer
 )
 
-# English model remains unchanged
+# Load the English sentiment model
 english_model = pipeline("sentiment-analysis", model="siebert/sentiment-roberta-large-english")
 
 class SentimentRequest(BaseModel):