WillHeld commited on
Commit
48086f4
·
verified ·
1 Parent(s): 3908642

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -21
app.py CHANGED
@@ -9,26 +9,9 @@ from datasets import Dataset
9
  from huggingface_hub import HfApi, login
10
  import time
11
 
12
- # Install required packages if not present
13
- try:
14
- from gradio_modal import Modal
15
- except ImportError:
16
- os.system('pip install gradio-modal')
17
- from gradio_modal import Modal
18
-
19
- # Install huggingface_hub if not present
20
- try:
21
- import huggingface_hub
22
- except ImportError:
23
- os.system('pip install huggingface_hub')
24
- import huggingface_hub
25
-
26
- # Install datasets if not present
27
- try:
28
- import datasets
29
- except ImportError:
30
- os.system('pip install datasets')
31
- import datasets
32
 
33
  # Model setup
34
  checkpoint = "WillHeld/soft-raccoon"
@@ -37,7 +20,7 @@ tokenizer = AutoTokenizer.from_pretrained(checkpoint)
37
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
38
 
39
  # Constants for dataset
40
- DATASET_REPO = "your-username/model-feedback-dataset" # Replace with your username
41
  DATASET_PATH = "./feedback_data" # Local path to store feedback
42
  DATASET_FILENAME = "feedback.jsonl" # Filename for feedback data
43
 
 
9
  from huggingface_hub import HfApi, login
10
  import time
11
 
12
+ from gradio_modal import Modal
13
+ import huggingface_hub
14
+ import datasets
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
 
16
  # Model setup
17
  checkpoint = "WillHeld/soft-raccoon"
 
20
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
21
 
22
  # Constants for dataset
23
+ DATASET_REPO = "WillHeld/model-feedback" # Replace with your username
24
  DATASET_PATH = "./feedback_data" # Local path to store feedback
25
  DATASET_FILENAME = "feedback.jsonl" # Filename for feedback data
26