attilasimko commited on
Commit
e47b5d6
·
1 Parent(s): ccf0698

weight code updated

Browse files
Files changed (1) hide show
  1. evaluations/weights.py +1 -1
evaluations/weights.py CHANGED
@@ -42,7 +42,7 @@ def evaluate(verbose, llm, zip, readme):
42
 
43
  if (llm):
44
  prompt = f"{readme}\nQ: Does this text contain a download link for the model pre-trained weights?"
45
- ans = model_predict(prompt)
46
  if (("Yes" in ans) & ("No" not in ans)):
47
  log(verbose, "LOG", "The LLM found signs for accessing the pre-trained weights from the readme")
48
  overall = "Yes"
 
42
 
43
  if (llm):
44
  prompt = f"{readme}\nQ: Does this text contain a download link for the model pre-trained weights?"
45
+ ans = llm.predict("STRICT", prompt)
46
  if (("Yes" in ans) & ("No" not in ans)):
47
  log(verbose, "LOG", "The LLM found signs for accessing the pre-trained weights from the readme")
48
  overall = "Yes"