Spaces:

Orawan
/

Task2

Sleeping

Orawan commited on Oct 6, 2023

Commit

6b80aca

•

1 Parent(s): 5f0cb84

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,8 @@
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained("zfox/finetuning-sentiment-model-3000-samples")
-model = AutoModelForSequenceClassification.from_pretrained("zfox/finetuning-sentiment-model-3000-samples")
 # Use a pipeline as a high-level helper
 from transformers import pipeline

 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+tokenizer = AutoTokenizer.from_pretrained("OatNapat/finetuned_yelp")
+model = AutoModelForSequenceClassification.from_pretrained("OatNapat/finetuned_yelp")
 # Use a pipeline as a high-level helper
 from transformers import pipeline