File size: 373 Bytes
7b83c9e
 
 
 
 
 
 
 
72a373d
7b83c9e
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import ollama
from huggingface_hub import login

def run_ollama(prompt,token=""):
    # Login to Hugging Face
    login(token=token)
    
    # Load and run the model
    model = ollama.load_model("tinyllama")
    response = model.run("Why is the sky blue")
    print(response)
    return response

if __name__ == "__main__":
    response = run_ollama()
    print(response)