maviced commited on
Commit
39fc371
·
verified ·
1 Parent(s): 144e6a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -16,18 +16,18 @@ import gradio as gr
16
  #model = learn.model
17
  #model = model.cpu()
18
 
19
- repo_id = "maviced/practica7-2"
20
- learn = from_pretrained_fastai(repo_id)
21
- model = learn.model
22
- model = model.cpu()
23
 
24
  #classifier = pipeline('text-classification', model='maviced/clasificador-rotten-tomatoes')
25
 
26
  def predict(text):
27
- pred,pred_idx,probs = model.predict(text)
28
- return pred
29
 
30
- #return classifier(text)[0]['label']
31
 
32
  gr.Interface(fn=predict, inputs="text", outputs="text",examples=['A visually stunning and emotionally resonant film that leaves a lasting impact.',
33
  'The film, while visually stunning, falls short with a predictable plot and underdeveloped characters, leaving the audience craving more substance.']).launch(share=True)
 
16
  #model = learn.model
17
  #model = model.cpu()
18
 
19
+ #repo_id = "maviced/practica7-2"
20
+ #learn = from_pretrained_fastai(repo_id)
21
+ #model = learn.model
22
+ #model = model.cpu()
23
 
24
  #classifier = pipeline('text-classification', model='maviced/clasificador-rotten-tomatoes')
25
 
26
  def predict(text):
27
+ #pred,pred_idx,probs = model.predict(text)
28
+ #return pred
29
 
30
+ return classifier(text)[0]['label']
31
 
32
  gr.Interface(fn=predict, inputs="text", outputs="text",examples=['A visually stunning and emotionally resonant film that leaves a lasting impact.',
33
  'The film, while visually stunning, falls short with a predictable plot and underdeveloped characters, leaving the audience craving more substance.']).launch(share=True)