Spaces:

ccm
/

chat-with-publications

Running on Zero

ccm commited on Jul 21, 2024

Commit

56e3d21

verified ·

1 Parent(s): 936fa72

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -6,6 +6,7 @@ import numpy  # to work with vectors
 import pandas  # to work with pandas
 import json  # to work with JSON
 import datasets  # to load the dataset
 # Load the dataset and convert to pandas
 full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()

 import pandas  # to work with pandas
 import json  # to work with JSON
 import datasets  # to load the dataset
+import space  # for GPU
 # Load the dataset and convert to pandas
 full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()