dangermouse77 commited on
Commit
a8fa556
·
verified ·
1 Parent(s): 9492245

Update test_aqmodel.py

Browse files
Files changed (1) hide show
  1. test_aqmodel.py +1 -1
test_aqmodel.py CHANGED
@@ -7,7 +7,7 @@ if len(sys.argv)<2:
7
  sys.exit(1)
8
 
9
  # Define model path
10
- model_path = "/home/dm/chat/AQ/aq_model_b8" # Make sure this points to your saved directory
11
 
12
  # Load model and tokenizer
13
  model = T5ForConditionalGeneration.from_pretrained(model_path)
 
7
  sys.exit(1)
8
 
9
  # Define model path
10
+ model_path = "./aq_model_b8" # Make sure this points to your saved directory
11
 
12
  # Load model and tokenizer
13
  model = T5ForConditionalGeneration.from_pretrained(model_path)