pbrusola commited on
Commit
22540a4
1 Parent(s): a744802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -1,15 +1,16 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
5
 
6
- def predict(image):
7
- predictions = pipeline(image)
8
- return {p["label"]: p["score"] for p in predictions}
 
9
 
10
  gr.Interface(
11
- predict,
12
- inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
13
- outputs=gr.outputs.Label(num_top_classes=2),
14
- title="Hot Dog? Or Not?",
15
  ).launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ nlp= pipeline(task="question-answering", model="mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
5
 
6
+ def qnamodel(context,question):
7
+ question_set = {'context':context,'question':question}
8
+ results = nlp(question_set)
9
+ return results['answer']
10
 
11
  gr.Interface(
12
+ qnamodel,
13
+ inputs=[gr.inputs.Textbox(label="Pon el texto a analizar aquí", max_lines=300), gr.inputs.Textbox(lines=2, label="Pregunta")],
14
+ outputs=gr.outputs.Textbox(label="Respuesta"),
15
+ title="Preguntas y respuestas",
16
  ).launch()