Spaces:

nlmaldonadog
/

Practica-8-sec2sec

Sleeping

nlmaldonadog commited on May 11

Commit

fc73c91

•

1 Parent(s): 739cdf9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 # Cargar el modelo
-model = pipeline('sec2sec', model='nlmaldonadog/mbart-clarification-P8')
 def predict(text):
     return model(text)[0]['generated_text']

 from transformers import pipeline
 # Cargar el modelo
+model = pipeline('text2text-generation', model='nlmaldonadog/mbart-clarification-P8')
 def predict(text):
     return model(text)[0]['generated_text']