Spaces:

fibleep
/

mickiewicz

Runtime error

Create app.py

2e4d3aa verified over 2 years ago

278 Bytes

import gradio as gr

from transformers import pipeline

pipe = pipeline(model="fibleep/mistral-7b-mickiewicz-finetune")


def predict(text):
    return pipe(text)[0]["generated_text"]


demo = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
)

demo.launch()

	import gradio as gr

	from transformers import pipeline

	pipe = pipeline(model="fibleep/mistral-7b-mickiewicz-finetune")


	def predict(text):
	return pipe(text)[0]["generated_text"]


	demo = gr.Interface(
	fn=predict,
	inputs="text",
	outputs="text",
	)

	demo.launch()