Spaces:
Running
Running
File size: 953 Bytes
e31a853 83083b0 cdf7ed2 e31a853 83083b0 e31a853 83083b0 cdf7ed2 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 | from ring import LLM_GPT35
from ring import LLM_GPT4
from ring import LLM_GPT4o
from ring import LLM_LLAMA2
from ring import LLM_GEMINI
from ring import LLM_AI21
from ring import PREF_ORDER_LLMS
from ring import generate_response
import sys
question = "What is the most common type of neuron in the brain?"
llm_ver = LLM_GPT4o
if "-g" in sys.argv:
llm_ver = LLM_GEMINI
if "-l" in sys.argv:
llm_ver = LLM_LLAMA2
if "-a" in sys.argv:
llm_ver = LLM_AI21
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()
|