Spaces:

huggingface
/

bloom-test-flax

Runtime error

sanchit-gandhi HF staff commited on Jul 13, 2022

Commit

0ac303f

•

1 Parent(s): ede6cee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=True):
         BOX_COMP: "#120F25",
     }
-    new_string = data[0]['generated_text']
     _, img = main(
         input_sentence,

         BOX_COMP: "#120F25",
     }
+    new_string = data[0]['generated_text'].split(input_sentence, 1)[1]
     _, img = main(
         input_sentence,