vhpvmx commited on
Commit
b285cb0
·
1 Parent(s): f298973

2da prueba con el model fine tuned

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,10 +6,10 @@ from transformers import pipeline
6
  #api = gr.Interface.load("./models/mt5-small-finetuned-amazon-en-es")
7
  #api = gr.Interface.load("huggingface/vhpvmx/mt5-small-finetuned-amazon-en-es") #ESTE LO CARGA
8
 
9
- # hub_model_id = "vhpvmx/mt5-small-finetuned-amazon-en-es"
10
- # summarizer = pipeline("summarization", model=hub_model_id)
11
 
12
- summarizer = pipeline("summarization")
13
 
14
  #model="./models/mt5-small-finetuned-amazon-en-es"
15
  #summarizer = pipeline("summarization", model)
 
6
  #api = gr.Interface.load("./models/mt5-small-finetuned-amazon-en-es")
7
  #api = gr.Interface.load("huggingface/vhpvmx/mt5-small-finetuned-amazon-en-es") #ESTE LO CARGA
8
 
9
+ hub_model_id = "vhpvmx/mt5-small-finetuned-amazon-en-es"
10
+ summarizer = pipeline("summarization", model=hub_model_id)
11
 
12
+ #summarizer = pipeline("summarization")
13
 
14
  #model="./models/mt5-small-finetuned-amazon-en-es"
15
  #summarizer = pipeline("summarization", model)