| import subprocess | |
| import sys | |
| subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "torch", "gradio"]) | |
| import gradio as gr | |
| from transformers import pipeline | |
| pipe = pipeline("text-generation", model="sapientinc/HRM-Text-1B") | |
| def chat(message, history): | |
| return pipe(message, max_length=200)[0]["generated_text"] | |
| gr.ChatInterface(fn=chat).launch() | |