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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -52,8 +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
-
57
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
58
 
59
 
@@ -77,5 +76,7 @@ examples = [
77
  ["los derivados de la serie LCG presentaron actividad citotóxica contra las líneas celulares SK-LU-1 y SW620"]
78
  ]
79
 
80
-
81
- 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 (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()