Spaces:

Babelscape
/

rebel-demo

Running

PereLluis13 commited on Oct 31, 2021

Commit

b6980f9

1 Parent(s): 282e7dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def load_models():
         _ = model.to("cuda:0") # comment if no GPU available
     _ = model.eval()
     print("+++++ loaded model", time() - st_time)
-    dataset = load_dataset('Babelscape/rebel-dataset', split="validation", streaming=True)
     return (tokenizer, model, dataset)
 def extract_triplets(text):

         _ = model.to("cuda:0") # comment if no GPU available
     _ = model.eval()
     print("+++++ loaded model", time() - st_time)
+    dataset = load_dataset('Babelscape/rebel-dataset', split="validation")
     return (tokenizer, model, dataset)
 def extract_triplets(text):