Grok-test-2 / app.py
Omnibus's picture
Create app.py
497ea18 verified
raw
history blame
154 Bytes
import torch
from transformers import pipeline
pipe = pipeline("text-generation", model="keyfan/grok-1-hf", trust_remote_code=True)
print(pipe("hello?"))