feat(project): update

Author: tpoisonooo

README.md

@@ -36,9 +36,13 @@ Furthermore, GraphGen incorporates multi-hop neighborhood sampling to capture co
 
 ### Gradio Demo
 
+1. Enter `webui` directory
    ```bash
-   pip install -r requirements.txt
-   gradio webui/app.py
+   cd webui
+   ```
+2. Run the Gradio demo
+   ```bash
+   gradio app.py
    ```
 
 ![ui](resources/images/interface.jpg)

evaluate.py

@@ -5,11 +5,6 @@
 import argparse
 import pandas as pd
 from dotenv import load_dotenv
-try:
-    import torch
-    import torch.multiprocessing as mp
-except Exception as e:
-    print(f'Please install torch and torch.multiprocessing to use {__file__}')
 from models import LengthEvaluator, MTLDEvaluator, RewardEvaluator, TextPair, UniEvaluator
 from utils import logger, set_logger
 
@@ -74,11 +69,13 @@ def evaluate_uni(corpus, uni_model_name):
 
 
 def clean_gpu_cache():
+    import torch
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
 
 if __name__ == '__main__':
+    import torch.multiprocessing as mp
     parser = argparse.ArgumentParser()
 
     parser.add_argument('--folder', type=str, default='cache/data', help='folder to load data')

graphgen/__init__.py

@@ -1,2 +0,0 @@
-from .version import __version__
-from .graphgen import GraphGen

requirements.txt

@@ -1,6 +1,6 @@
 tqdm
 openai
-python - dotenv
+python-dotenv
 numpy
 networkx
 graspologic