Spaces:

huggingface
/

bloom-test-flax

Runtime error

valhalla commited on Jul 13, 2022

Commit

b4f5897

1 Parent(s): f771a11

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
     )
     if raw_text:
-        return None, data[0]['generated_text']
     width, height = 3326, 3326
     assets_path = "assets"

     )
     if raw_text:
+        return None, data['generated_text'][0]
     width, height = 3326, 3326
     assets_path = "assets"