main.py 141 Bytes
Newer Older
Jeffrey Morgan's avatar
Jeffrey Morgan committed
1
from langchain.llms import Ollama
2
3

input = input("What is your question?")
Jeffrey Morgan's avatar
Jeffrey Morgan committed
4
llm = Ollama(model="llama2")
5
res = llm.predict(input)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
6
print (res)