3ie-intervention-outcome-entity-linking

Sleeping

CarlosMalaga commited on May 3

Commit

64f0995

•

1 Parent(s): 6ea46ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -166,9 +166,9 @@ def load_model():
                 ),
                 metadata_fields=["definition"],
                 separator=' <def> ',
-                device="cuda"
             ),
-            devide="cuda"
     )
     retriever.index()

                 ),
                 metadata_fields=["definition"],
                 separator=' <def> ',
+                device="cpu"
             ),
+            devide="cpu"
     )
     retriever.index()