Nfanlo commited on
Commit
247d1f4
1 Parent(s): e490a50

change app works

Browse files
Files changed (1) hide show
  1. 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
- "clasificaci贸n-de-texto",
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
- "clasificaci贸n-de-texto",
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,