Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 import torch
 import nltk
 nltk.download("punkt")
-from transformers import BertTokenizerFast, BertTokenizer, BertForTokenClassification, BertForSequenceClassification, Pipeline
+from transformers import ConvBertTokenizerFast, ConvBertForTokenClassification, BertTokenizer, BertForTokenClassification, BertForSequenceClassification, Pipeline
 from nltk import sent_tokenize
 import uvicorn
 from fastapi import FastAPI
@@ -116,8 +116,8 @@ class AspectSentimentPipeline(Pipeline):
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-aspect_extraction_model = BertForTokenClassification.from_pretrained("thealper2/aspect-extraction-model")
-aspect_extraction_tokenizer = BertTokenizerFast.from_pretrained("thealper2/aspect-extraction-tokenizer")
+aspect_extraction_model = ConvBertForTokenClassification.from_pretrained("thealper2/aspect-extraction-model")
+aspect_extraction_tokenizer = ConvBertTokenizerFast.from_pretrained("thealper2/aspect-extraction-tokenizer")
 
 aspect_sentiment_model = BertForSequenceClassification.from_pretrained("thealper2/aspect-sentiment-model")
 aspect_sentiment_tokenizer = BertTokenizer.from_pretrained("thealper2/aspect-sentiment-tokenizer")