Spaces:
Runtime error
Runtime error
import gradio as gr | |
import requests | |
import os | |
##Bloom Inference API | |
API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom" | |
HF_TOKEN = os.environ["HF_TOKEN"] | |
headers = {"Authorization": f"Bearer {HF_TOKEN}"} | |
def write_essay(sample_topic, user_topic): | |
if len(sample_topic) == 0: | |
sample_topic = user_topic | |
p = "Instruction: Write a five-paragraph essay for the following topic:\nSuggested Length: more than 120 words\nTOPIC: " + sample_topic + "\n\nESSAY:" | |
try: | |
json_ = {"inputs": p, | |
"parameters": | |
{ | |
"top_p": 0.9, | |
"temperature": 1.1, | |
"max_new_tokens": 250, | |
"return_full_text": True | |
}, "options": | |
{ | |
"use_cache": True, | |
"wait_for_model":True | |
},} | |
response = requests.post(API_URL, headers=headers, json=json_) | |
print(f"Response is : {response}") | |
output = response.json() | |
print(f"output is : {output}") #{output}") | |
output_tmp = output[0]['generated_text'] | |
print(f"output_tmp is: {output_tmp}") | |
#print(response) | |
return output_tmp | |
except: | |
return "Invalid Query" | |
demo = gr.Blocks() | |
with demo: | |
gr.Markdown("<h3><center>人工知能から学ぶ、英検作文へのヒント</center></h3>") | |
gr.Markdown("<h4><center>使い方:トピックを選んで画面左下にある「エッセイ作成」ボタンをクリック</center></h4>") | |
gr.Markdown("<h4><center>※重要:費用があまりにも大きくなってきましたのでサービスを中止させていただきます。</center></h4>") | |
gr.Markdown("<h4><center>専用のAPIキーを取得してご利用ください。</center></h4>") | |
gr.Markdown("<center>Created by Choimirai School</center>") | |
with gr.Row(): | |
sample_topic = gr.Radio([ | |
"Will humans live on other planets someday?", | |
"Japan should become a completely cashless society.", | |
"Global overpopulation is a serious threat to the future of humankind.", | |
"Improving relations with other Asian nations should be a priority for the Japanese government.", | |
"Can renewable energy sources replace fossil fuels?", | |
"Should democratic nations actively promote the spread of democracy to nondemocratic nations?", | |
"Agree or disagree, Infectious diseases will become a bigger problem in the coming decades.", ], label= "Choose a sample TOPIC") | |
user_topic = gr.Textbox(label="Or, write your own topic for Eiken essay.", value="Will fossil fuels such as oil and gas still be the world's main source of energy in the coming decades?") | |
with gr.Row(): | |
written_essay = gr.inputs.Textbox(lines=20, label="Sample Essay written by GPT-3") | |
b1 = gr.Button("エッセイ作成") | |
b1.click(write_essay, inputs = [sample_topic, user_topic], outputs = written_essay) | |
with gr.Row(): | |
gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=sangmin_eiken-essay-with-gpt3)") | |
demo.launch(enable_queue=True, debug=True) | |