Spaces:
Running
Running
aisyahhrazak
commited on
Commit
•
c86a287
1
Parent(s):
8ce4957
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ nltk.download('punkt_tab', quiet=True)
|
|
33 |
nltk.download('stopwords', quiet=True)
|
34 |
nltk.download('vader_lexicon', quiet=True)
|
35 |
tokenizer_tpb = AutoTokenizer.from_pretrained('mesolitica/malaysian-mistral-191M-MLM-512')
|
36 |
-
model_tpb = MistralForSequenceClassification.from_pretrained('HalalFoodNLP/tpb-model-halal', torch_dtype=torch.bfloat16)
|
37 |
model_sentiment = MistralForSequenceClassification.from_pretrained('malaysia-ai/sentiment-mistral-191M-MLM', torch_dtype=torch.bfloat16)
|
38 |
pipeline_tpb = pipeline(task="text-classification", model=model_tpb, tokenizer=tokenizer_tpb)
|
39 |
sentiment_pipeline = pipeline("sentiment-analysis", model=model_sentiment, tokenizer=tokenizer_tpb)
|
|
|
33 |
nltk.download('stopwords', quiet=True)
|
34 |
nltk.download('vader_lexicon', quiet=True)
|
35 |
tokenizer_tpb = AutoTokenizer.from_pretrained('mesolitica/malaysian-mistral-191M-MLM-512')
|
36 |
+
model_tpb = MistralForSequenceClassification.from_pretrained('HalalFoodNLP/tpb-model-halal', torch_dtype=torch.bfloat16, token = HF_TOKEN)
|
37 |
model_sentiment = MistralForSequenceClassification.from_pretrained('malaysia-ai/sentiment-mistral-191M-MLM', torch_dtype=torch.bfloat16)
|
38 |
pipeline_tpb = pipeline(task="text-classification", model=model_tpb, tokenizer=tokenizer_tpb)
|
39 |
sentiment_pipeline = pipeline("sentiment-analysis", model=model_sentiment, tokenizer=tokenizer_tpb)
|