fix streaming

app.py

@@ -22,7 +22,8 @@ def load_models():
     _ = model.eval()
     print("+++++ loaded model", time() - st_time)
     dataset = load_dataset('Babelscape/rebel-dataset', split="validation", streaming=True)
-    return (tokenizer, model, dataset.take(1000))
+    dataset = [example for example in dataset.take(1001)]
+    return (tokenizer, model, dataset)
 
 def extract_triplets(text):
     triplets = []