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")
|