thealper2 commited on
Commit
67ca456
·
verified ·
1 Parent(s): e99e60a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -114,11 +114,11 @@ class AspectSentimentPipeline(Pipeline):
114
 
115
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
116
 
117
- aspect_extraction_model = BertForTokenClassification.from_pretrained("thealper2/aspect_extraction_model")
118
- aspect_extraction_tokenizer = BertTokenizerFast.from_pretrained("thealper2/aspect_extraction_tokenizer")
119
 
120
- aspect_sentiment_model = BertForSequenceClassification.from_pretrained("thealper2/aspect_sentiment_model")
121
- aspect_sentiment_tokenizer = BertTokenizer.from_pretrained("thealper2/aspect_sentiment_tokenizer")
122
 
123
  pipeline = AspectSentimentPipeline(
124
  aspect_extraction_model=aspect_extraction_model,
 
114
 
115
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
116
 
117
+ aspect_extraction_model = BertForTokenClassification.from_pretrained("thealper2/aspect-extraction-model")
118
+ aspect_extraction_tokenizer = BertTokenizerFast.from_pretrained("thealper2/aspect-extraction-tokenizer")
119
 
120
+ aspect_sentiment_model = BertForSequenceClassification.from_pretrained("thealper2/aspect-sentiment-model")
121
+ aspect_sentiment_tokenizer = BertTokenizer.from_pretrained("thealper2/aspect-sentiment-tokenizer")
122
 
123
  pipeline = AspectSentimentPipeline(
124
  aspect_extraction_model=aspect_extraction_model,