edgilr commited on
Commit
3f46b35
verified
1 Parent(s): fca9848

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,24 +1,24 @@
1
  import pandas as pd
2
  from fastai.text.all import *
3
  from datasets import load_dataset
4
- #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
- 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')
15
 
16
  def predict(txt):
17
  # Modelo de clasificaci贸n usando LSTM o modelo de clasificaci贸n de texto usando modelos de lenguaje
18
- pred,pred_idx,probs = learner.predict(txt)
19
- return pred
20
  # Modelo de clasificaci贸n basados en mecanismos de atenci贸n
21
- #return classifier(txt)['label']
22
 
23
  gr.Interface(fn=predict, inputs="text", outputs="text",
24
  examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',
 
1
  import pandas as pd
2
  from fastai.text.all import *
3
  from datasets import load_dataset
4
+ 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
+ #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')
15
 
16
  def predict(txt):
17
  # Modelo de clasificaci贸n usando LSTM o modelo de clasificaci贸n de texto usando modelos de lenguaje
18
+ #pred,pred_idx,probs = learner.predict(txt)
19
+ #return pred
20
  # Modelo de clasificaci贸n basados en mecanismos de atenci贸n
21
+ return classifier(txt)['label']
22
 
23
  gr.Interface(fn=predict, inputs="text", outputs="text",
24
  examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',