Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline | |
model_id = "TheBloke/MythoMax-L2-13B-GGUF" | |
tokenizer = AutoTokenizer.from_pretrained(model_id) | |
model = AutoModelForCausalLM.from_pretrained(model_id) | |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer) | |
def chat(prompt): | |
response = pipe(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)[0]["generated_text"] | |
return response | |
gr.ChatInterface(chat).launch() | |