Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
 from transformers import pipeline
 
-def convert(Sentence):
+def convert(prompt):
     model = pipeline('text-generation', model='bigscience/bloom-1b1')
-    result = model(Sentence)
-    return result[0]
+    result = model(prompt, max_length=40)
+    return result[0]['generated_text']
 
 with gr.Blocks() as bls:
     gr.Markdown("Here is a Text Generation app")
     with gr.Row():
         inp = gr.Textbox(label="Type your prompt here and click Run", placeholder='Example: The main cloud services of AWS are:')
-        out = gr.Textbox(label="The sentiment is:")
+        out = gr.Textbox(label="Output")
     btn = gr.Button("Run")
     btn.click(fn=convert, inputs=inp, outputs=out)