Spaces:
Running
Running
import os | |
import torch | |
from huggingface_hub import HfApi | |
# replace this with our token | |
# TOKEN = os.environ.get("HF_TOKEN", None) | |
TOKEN = os.getenv("H4_TOKEN1") | |
TOKEN1 = os.getenv("H4_TOKEN1") | |
TOKEN2 = os.getenv("H4_TOKEN2") | |
TOKEN3 = os.getenv("H4_TOKEN3") | |
TOKEN4 = os.getenv("H4_TOKEN4") | |
TOKEN5 = os.getenv("H4_TOKEN5") | |
TOKEN6 = os.getenv("H4_TOKEN6") | |
TOKEN7 = os.getenv("H4_TOKEN7") | |
TOKEN8 = os.getenv("H4_TOKEN8") | |
TOKEN9 = os.getenv("H4_TOKEN9") | |
TOKEN10 = os.getenv("H4_TOKEN10") | |
TOKEN11 = os.getenv("H4_TOKEN11") | |
TOKEN12 = os.getenv("H4_TOKEN12") | |
TOKEN13 = os.getenv("H4_TOKEN13") | |
TOKEN14 = os.getenv("H4_TOKEN14") | |
TOKEN15 = os.getenv("H4_TOKEN15") | |
TOKEN16 = os.getenv("H4_TOKEN16") | |
TOKEN17 = os.getenv("H4_TOKEN17") | |
TOKEN18 = os.getenv("H4_TOKEN18") | |
TOKEN19 = os.getenv("H4_TOKEN19") | |
TOKEN20 = os.getenv("H4_TOKEN20") | |
# print("H4_token:", TOKEN) | |
OWNER = "Simondon" # Change to your org - don't forget to create a results and request dataset, with the correct format! | |
# ---------------------------------- | |
REPO_ID = f"{OWNER}/HumanLikeness" | |
QUEUE_REPO = f"{OWNER}/requests" | |
RESULTS_REPO = f"{OWNER}/results" | |
# print(RESULTS_REPO) | |
CACHE_PATH=os.getenv("HF_HOME", ".") | |
# Local caches | |
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue") | |
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results") | |
EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk") | |
EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk") | |
# print(EVAL_RESULTS_PATH) | |
# exit() | |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu') #"cpu" | |
API = HfApi(token=TOKEN) | |
DATASET_PATH = "./src/datasets/Material_Llama2_0603.xlsx" #experiment data | |
PROMPT_PATH = "./src/datasets/prompt.xlsx" #prompt for each experiment | |
HEM_PATH = 'vectara/hallucination_evaluation_model' | |
HUMAN_DATA = "./src/datasets/human_data_coding.csv" #experiment data | |
ITEM_4_DATA = "./src/datasets/associataion_dataset.csv" #database | |
ITEM_5_DATA = "./src/datasets/Items_5.csv" #experiment 5 need verb words | |
# SYSTEM_PROMPT = "You are a chat bot answering questions using data. You must stick to the answers provided solely by the text in the passage provided." | |
SYSTEM_PROMPT = "You are participating in a psycholinguistic experiment. You will complete a task on English language use. Please respond to the questions directly, without using introductory phrases (e.g., Sure or OK) or special formats at the beginning of your responses." | |
'''prompt''' | |
# USER_PROMPT = "You are asked the question 'Provide a concise summary of the following passage, covering the core pieces of information described': " | |
USER_PROMPT = "" |