cdb24 commited on
Commit
075edf9
1 Parent(s): 5aface3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ import streamlit as st
9
  from streamlit_chat import message
10
  import requests
11
 
12
- model_one = "distilbert-base-uncased-finetuned-sst-2-english"
13
- model_two = "Newtral/xlm-r-finetuned-toxic-political-tweets-es"
14
 
15
  def toxicRating(text, model):
16
  model = AutoModelForSequenceClassification.from_pretrained(model)
 
9
  from streamlit_chat import message
10
  import requests
11
 
12
+ model_one = "distilbert-base-uncased-finetuned-sst-2-english"
13
+ model_two = "Newtral/xlm-r-finetuned-toxic-political-tweets-es"
14
 
15
  def toxicRating(text, model):
16
  model = AutoModelForSequenceClassification.from_pretrained(model)