Spaces:
Runtime error
Runtime error
Commit
·
93184b1
1
Parent(s):
5d4d985
updt runllm
Browse files- run_llm.py +6 -0
run_llm.py
CHANGED
|
@@ -464,8 +464,14 @@ def run_llm_interface(model_path, prompt, sentence):
|
|
| 464 |
)
|
| 465 |
|
| 466 |
# Run the main function
|
|
|
|
|
|
|
| 467 |
main(args=args)
|
| 468 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 469 |
if __name__ == "__main__":
|
| 470 |
parser = argparse.ArgumentParser()
|
| 471 |
add_model_args(parser)
|
|
|
|
| 464 |
)
|
| 465 |
|
| 466 |
# Run the main function
|
| 467 |
+
# For simplicity, assuming prompt values 1, 2, and 3 correspond to different strategies
|
| 468 |
+
# You may need to adjust this based on your actual logic
|
| 469 |
main(args=args)
|
| 470 |
|
| 471 |
+
# Return dummy values for now, replace with actual outputs
|
| 472 |
+
return "Strategy 1 Output", "Strategy 2 Output", "Strategy 3 Output"
|
| 473 |
+
|
| 474 |
+
|
| 475 |
if __name__ == "__main__":
|
| 476 |
parser = argparse.ArgumentParser()
|
| 477 |
add_model_args(parser)
|