feat: Added application file

src/api.py

@@ -2,8 +2,33 @@ import configparser
 
 from fastapi import FastAPI
 from pydantic import BaseModel
-from SentimentAndIntentionAnalysis import ZeroShotClassifier
 from data_loader import *
+from transformers import pipeline, BartTokenizer, BartForSequenceClassification
+class ZeroShotClassifier:
+
+    def __init__(self, model_name, sentiment_labels, intention_labels):
+        self.model = self.create_model(model_name)
+        self.model_name = model_name
+        self.sentiment_labels = sentiment_labels
+        self.intention_labels = intention_labels
+
+    def create_model(self, model_name):
+        # Create Model
+        tokenizer = BartTokenizer.from_pretrained(model_name)
+        model = BartForSequenceClassification.from_pretrained(model_name)
+        classifier = pipeline("zero-shot-classification", model=model, tokenizer=tokenizer)
+        return classifier
+
+    def analyze_text(self, text):
+        # Sentiment analysis
+        sentiment_result = self.model(text, self.sentiment_labels)
+        sentiment = sentiment_result["labels"][0]
+
+        # Intention analysis
+        intention_result = self.model(text, self.intention_labels)
+        intention = intention_result["labels"][0]
+
+        return {"sentiment": sentiment, "intention": intention}
 
 # Initialize FastAPI app
 app = FastAPI()