Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import pipeline
 
 def convert(prompt):
-    model = pipeline('text-generation', model='bigscience/bloom-560m')
-    result = model(prompt, max_length=400, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, repetition_penalty = 2.0, diversity_penalty = 0.9)
+    model = pipeline('text-generation', model='bigscience/bloom-1b1')
+    result = model(prompt, max_length=100, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, repetition_penalty = 2.0, diversity_penalty = 0.9)
     return result[0]['generated_text']
 
 with gr.Blocks() as bls: