Spaces:
Running
Running
| import os | |
| import requests | |
| import gradio as gr | |
| # ENV vars | |
| API_URL = os.environ["API_URL"] | |
| HF_TOKEN = os.environ["HF_TOKEN"] | |
| headers = {"Authorization": f"Bearer {HF_TOKEN}"} | |
| comment_syntaxes = { | |
| "C": "/* {} */", | |
| "C++": "/* {} */", | |
| "Java": "/* {} */", | |
| "Golang": "/* {} */", | |
| "Rust": "/* {} */", | |
| "Javascript": "/* {} */", | |
| "PHP": "/* {} */", | |
| "Kotlin": "/* {} */", | |
| "HTML": "<!-- {} -->", | |
| "Python": "#{}", | |
| "Bash": "#{}", | |
| "Ruby": "=begin {} =end", | |
| } | |
| def get_script(lang, instruction): | |
| jsn = {"inputs": comment_syntaxes[lang].format("instruction: " + instruction), | |
| "parameters": | |
| { | |
| "top_p": 0.9, | |
| "max_new_tokens": 64, | |
| "return_full_text": True, | |
| "do_sample": True, | |
| }, | |
| "options": | |
| {"use_cache": True, | |
| "wait_for_model": True, | |
| }, } | |
| response = requests.post(API_URL, headers=headers, json=jsn) | |
| return response.json()["generated_text"] | |
| demo = gr.Blocks() | |
| with demo: | |
| gr.Markdown( | |
| "<h1><center>Give Instructions to Generate a Program</center></h1>") | |
| with gr.Row(): | |
| dropdown = gr.Dropdown( | |
| choices=comment_syntaxes.keys(), label="Choose the language") | |
| # with gr.Column: | |
| instruction = gr.Textbox(label="Write a instruction", | |
| value="Create a python function that generates random password with given length with ascii characters ", lines=6) | |
| with gr.Row(): | |
| generated_txt = gr.Textbox(lines=5) | |
| btn = gr.Button("Generate") | |
| btn.click(get_script, inputs=[dropdown, | |
| instruction], outputs=generated_txt) | |
| with gr.Row(): | |
| gr.Markdown( | |
| "") | |
| demo.launch(enable_queue=True, debug=True) | |