Spaces:

azaninello
/

ailai

Runtime error

ailai / app.py

Update app.py

5ad9463 almost 2 years ago

No virus

517 Bytes

	import gradio as gr
	import transformers
	from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

	shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('azaninello/gpt2-finetuned-shrooms'), tokenizer="gpt2")

	def generator(scrivi_un_titolo = ""):
	shroom_result = shroom_generator(f"<\|HEADLINE\|>{scrivi_un_titolo}<\|TEXT\|>", max_length=420)
	return shroom_result[0]["generated_text"]

	iface = gr.Interface(fn=generator, inputs="text", outputs="text")
	iface.launch()