clavel commited on
Commit
d336538
1 Parent(s): 234502e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -52,7 +52,7 @@ tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classificatio
52
  model = AutoModelForSequenceClassification.from_pretrained(
53
  'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
54
  #pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
55
- pipe = pipeline (model='hackathon-pln-es/unam_tesis_BETO_finnetuning')
56
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
57
 
58
 
@@ -76,7 +76,7 @@ examples = [
76
  ["los derivados de la serie LCG presentaron actividad citotóxica contra las líneas celulares SK-LU-1 y SW620"]
77
  ]
78
 
79
- gr.Interface(fn=thesis_prediction, inputs=[gr.inputs.Textbox(label="Título o frase:", placeholder="Ingrese de favor el título de la tesis o un fragmento de esta."),], outputs=[gr.outputs.HTML(label="La tesis pertenece a la carrera de: ",)], title=title, description=description, article=article, examples=examples).launch()
80
  #gr.Interface(fn=thesis_prediction, inputs=[gr.inputs.Textbox(label="Título o frase:", placeholder="Ingrese de favor el título de la tesis o un fragmento de esta."),], outputs="keyClass", title=title, description=description, article=article, examples=examples).launch()
81
  #gr.Interface.load('huggingface/hackathon-pln-es/unam_tesis_BETO_finnetuning', placeholder="Ingrese de favor el título de la tesis o un fragmento de esta.", title=title, description=description, article=article, examples=examples).launch()
82
- #gr.Interface.from_pipeline(pipe, title=title, description=description, article=article, examples=examples).launch()
 
52
  model = AutoModelForSequenceClassification.from_pretrained(
53
  'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
54
  #pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
55
+ pipe = pipeline ("text-classification", model='hackathon-pln-es/unam_tesis_BETO_finnetuning')
56
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
57
 
58
 
 
76
  ["los derivados de la serie LCG presentaron actividad citotóxica contra las líneas celulares SK-LU-1 y SW620"]
77
  ]
78
 
79
+ #gr.Interface(fn=thesis_prediction, inputs=[gr.inputs.Textbox(label="Título o frase:", placeholder="Ingrese de favor el título de la tesis o un fragmento de esta."),], outputs=[gr.outputs.HTML(label="La tesis pertenece a la carrera de: ",)], title=title, description=description, article=article, examples=examples).launch()
80
  #gr.Interface(fn=thesis_prediction, inputs=[gr.inputs.Textbox(label="Título o frase:", placeholder="Ingrese de favor el título de la tesis o un fragmento de esta."),], outputs="keyClass", title=title, description=description, article=article, examples=examples).launch()
81
  #gr.Interface.load('huggingface/hackathon-pln-es/unam_tesis_BETO_finnetuning', placeholder="Ingrese de favor el título de la tesis o un fragmento de esta.", title=title, description=description, article=article, examples=examples).launch()
82
+ gr.Interface.from_pipeline(pipe, title=title, description=description, article=article, examples=examples).launch()