Spaces:

XquanL
/

hello-world

Runtime error

App Files Files Community

hello-world / app.py

XquanL's picture

Update app.py

b4fa516 over 1 year ago

raw history blame contribute delete

No virus

328 Bytes

	from transformers import pipeline
	import gradio as gr

	#define the model
	model = pipeline("text-generation")

	#define the predict function
	def predict(prompt):
	completion = model(prompt)[0]["generated_text"]
	return completion

	#define the gradio interface
	gr.Interface(fn=predict, inputs="text", outputs="text").launch()