Gregoryjr
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ if con:
|
|
25 |
score = result[0]['score']
|
26 |
st.write(f"Prediction: {label}, Score: {score*100}% chance")
|
27 |
|
28 |
-
if model == "Greys/milestonemodel"
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
31 |
model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
|
|
|
25 |
score = result[0]['score']
|
26 |
st.write(f"Prediction: {label}, Score: {score*100}% chance")
|
27 |
|
28 |
+
if model == "Greys/milestonemodel":
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
31 |
model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
|