MrDdz commited on
Commit
137b16f
1 Parent(s): 3f3a57e

model changed

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -74,12 +74,12 @@ def preprocess(text):
74
  @st.cache_resource()
75
  def get_model():
76
  # Load the model and tokenizer
77
- tokenizer = AutoTokenizer.from_pretrained("xlnet-base-cased")
78
- model = AutoModelForSequenceClassification.from_pretrained("MrDdz/mytuned_test_trainer-base-cased1")
79
  return tokenizer,model
80
  tokenizer, model = get_model()
81
 
82
- if text and button:
83
  text_sample = tokenizer(text, padding = 'max_length',return_tensors = 'pt')
84
  # print(text_sample)
85
  output = model(**text_sample)
 
74
  @st.cache_resource()
75
  def get_model():
76
  # Load the model and tokenizer
77
+ tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
78
+ model = AutoModelForSequenceClassification.from_pretrained("MrDdz/bert-base-uncased ")
79
  return tokenizer,model
80
  tokenizer, model = get_model()
81
 
82
+ if button:
83
  text_sample = tokenizer(text, padding = 'max_length',return_tensors = 'pt')
84
  # print(text_sample)
85
  output = model(**text_sample)