Spaces:
Running
Running
File size: 1,056 Bytes
a41e4c6 1667843 e31a853 83083b0 e31a853 a41e4c6 e31a853 a41e4c6 83083b0 cdf7ed2 1667843 83083b0 e31a853 83083b0 e31a853 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | from openworm_ai.utils.llms import LLM_GPT4o
from openworm_ai.utils.llms import LLM_LLAMA2
from openworm_ai.utils.llms import LLM_GEMINI_2F
from openworm_ai.utils.llms import LLM_GEMINI_25F
from openworm_ai.utils.llms import LLM_COHERE
from ring import generate_response
import sys
question = "What is the most common type of neuron in the brain?"
llm_ver = LLM_GPT4o
if "-g2" in sys.argv:
llm_ver = LLM_GEMINI_2F
if "-g25" in sys.argv:
llm_ver = LLM_GEMINI_25F
if "-l" in sys.argv:
llm_ver = LLM_LLAMA2
if "-co" in sys.argv:
llm_ver = LLM_COHERE
print("--------------------------------------------------------")
print("Asking question:\n %s" % question)
print("--------------------------------------------------------")
print(" ... Connecting to: %s" % llm_ver)
response = generate_response(question, llm_ver, temperature=0, only_celegans=False)
print("--------------------------------------------------------")
print("Answer:\n %s" % response)
print("--------------------------------------------------------")
print()
|