Update app.py
Browse files
app.py
CHANGED
|
@@ -5,10 +5,10 @@ from transformers import pipeline
|
|
| 5 |
import gradio as gr
|
| 6 |
|
| 7 |
# Modelo de clasificación usando LSTM
|
| 8 |
-
learner = load_learner('modelLSTM.pkl')
|
| 9 |
|
| 10 |
# Modelo de clasificación de texto usando modelos de lenguaje
|
| 11 |
-
|
| 12 |
|
| 13 |
# Modelo de clasificación basados en mecanismos de atención
|
| 14 |
#classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')
|
|
|
|
| 5 |
import gradio as gr
|
| 6 |
|
| 7 |
# Modelo de clasificación usando LSTM
|
| 8 |
+
#learner = load_learner('modelLSTM.pkl')
|
| 9 |
|
| 10 |
# Modelo de clasificación de texto usando modelos de lenguaje
|
| 11 |
+
learner = load_learner('modelML.pkl')
|
| 12 |
|
| 13 |
# Modelo de clasificación basados en mecanismos de atención
|
| 14 |
#classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')
|