Spaces:
Runtime error
Runtime error
File size: 849 Bytes
881ae41 c59b828 881ae41 c59b828 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
import requests
def generate(prompt = "", model = "bigscience/bloom", use_cache = False, **kwargs):
json = {"inputs": prompt, "parameters": kwargs, "options": {"use_cache": use_cache}}
response = requests.post(f"https://api-inference.huggingface.co/models/{model}", json=json)
try:
output = response.json()[0]["generated_text"].replace(prompt, "")
except:
output = ""
return output
def generate_line(story, temperature):
out = generate(story, temperature=temperature, max_new_tokens=50)
i = out.find("\n")
return story + out[:i]
with gr.Blocks() as demo:
story = gr.Textbox(value="Character_1:\n\nCharacter_2:", label="Story")
generate_button = gr.Button("Generate")
generate_button.click(generate_line, inputs=[story, gr.Slider(0, 2, 0.9, label="temperature")], outputs=[story])
demo.launch() |