Nfanlo
commited on
Commit
路
247d1f4
1
Parent(s):
e490a50
change app works
Browse files- showcase_app.py +2 -2
showcase_app.py
CHANGED
@@ -65,7 +65,7 @@ def update_model(
|
|
65 |
model_id = local_model_id
|
66 |
st.text(f"Cargando modelo {model_id}")
|
67 |
pipe = pipeline(
|
68 |
-
"
|
69 |
model=AutoModelForSequenceClassification.from_pretrained(model_id),
|
70 |
tokenizer=AutoTokenizer.from_pretrained(model_id),
|
71 |
return_all_scores=True,
|
@@ -78,7 +78,7 @@ def update_model(
|
|
78 |
for model in models:
|
79 |
with st.spinner(f"Cargando modelo {model}"):
|
80 |
pipe = pipeline(
|
81 |
-
"
|
82 |
model=AutoModelForSequenceClassification.from_pretrained(model),
|
83 |
tokenizer=AutoTokenizer.from_pretrained(model),
|
84 |
return_all_scores=True,
|
|
|
65 |
model_id = local_model_id
|
66 |
st.text(f"Cargando modelo {model_id}")
|
67 |
pipe = pipeline(
|
68 |
+
"text-classification",
|
69 |
model=AutoModelForSequenceClassification.from_pretrained(model_id),
|
70 |
tokenizer=AutoTokenizer.from_pretrained(model_id),
|
71 |
return_all_scores=True,
|
|
|
78 |
for model in models:
|
79 |
with st.spinner(f"Cargando modelo {model}"):
|
80 |
pipe = pipeline(
|
81 |
+
"text-classification",
|
82 |
model=AutoModelForSequenceClassification.from_pretrained(model),
|
83 |
tokenizer=AutoTokenizer.from_pretrained(model),
|
84 |
return_all_scores=True,
|