Hugging Face's logo

Spaces:
Preetham04
/
clothing_bert
Sleeping

App Files Community
Preetham04 commited on
Commit
df10d8f
•
1 Parent(s): 292a86c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import gradio as gr
11
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
12
 
13
  # Load the base model
14
- base_model_name = "bert-base-uncased"
15
  tokenizer = AutoTokenizer.from_pretrained(base_model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(base_model_name)
17
 
 
11
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
12
 
13
  # Load the base model
14
+ base_model_name = "Preetham04/sentiment-analysis"
15
  tokenizer = AutoTokenizer.from_pretrained(base_model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(base_model_name)
17