mixtral-test / app.py
Syksteaz's picture
Create app.py
4cbcfc5 verified
Raw History Blame Contribute Delete
375 Bytes
# Use a pipeline as a high-level helper
from transformers import pipeline
import gradio as gr
pipe = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")
def generate(input_text):
generated_text = pipe(input_text)
return generated_text
iface = gr.Interface(fn=generate, inputs=gr.Text(label='Your prompt here'), outputs="text")
iface.launch()