Spaces:

sandeepmajumdar
/

Bloom-Slim-Text-Generation

Build error

sandeepmajumdar commited on Aug 17, 2022

Commit

e7695c6

1 Parent(s): ddb3a8b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 def convert(prompt):
-    model = pipeline('text-generation', model='bigscience/bloom-1b1')
     result = model(prompt, max_length=40)
     return result[0]['generated_text']

 from transformers import pipeline
 def convert(prompt):
+    model = pipeline('text-generation', model='bigscience/bloom')
     result = model(prompt, max_length=40)
     return result[0]['generated_text']