Spaces:

sandeepmajumdar
/

Bloom-Slim-Text-Generation

Build error

sandeepmajumdar commited on Aug 17, 2022

Commit

6bc5c91

1 Parent(s): 9a9a4a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline
 def convert(Sentence):
     model = pipeline('text-generation', model='bigscience/bloom-1b1')
     result = model(Sentence)
-    return result[0]['label'])
 with gr.Blocks() as bls:
     gr.Markdown("Here is a Text Generation app")

 def convert(Sentence):
     model = pipeline('text-generation', model='bigscience/bloom-1b1')
     result = model(Sentence)
+    return result[0]
 with gr.Blocks() as bls:
     gr.Markdown("Here is a Text Generation app")