File size: 2,542 Bytes
bf3219e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import gradio as gr
# Load the models
models = {
"bigcode/python-stack-v1-functions-filtered-sc2-subset": gr.Interface.load("bigcode/python-stack-v1-functions-filtered-sc2-subset"),
"bigcode/python-stack-v1-functions-filtered-sc2": gr.Interface.load("bigcode/python-stack-v1-functions-filtered-sc2"),
"muellerzr/python-stack-v1-functions-filtered-llama-3-8B": gr.Interface.load("muellerzr/python-stack-v1-functions-filtered-llama-3-8B"),
"YoLo2000/python-stack-functions-filtered": gr.Interface.load("YoLo2000/python-stack-functions-filtered"),
"YoLo2000/python-stack-functions-filteredbigcode/python-stack-v1-functions-filtered-sc2": gr.Interface.load("YoLo2000/python-stack-functions-filteredbigcode/python-stack-v1-functions-filtered-sc2"),
"TheBloke/Python-Code-13B-GGUF": gr.Interface.load("TheBloke/Python-Code-13B-GGUF"),
"replit/replit-code-v1_5-3b": gr.Interface.load("replit/replit-code-v1_5-3b"),
"neulab/codebert-python": gr.Interface.load("neulab/codebert-python")
}
# Load the datasets
datasets = {
"kye/all-huggingface-python-code": gr.Dataset.load("kye/all-huggingface-python-code"),
"ajibawa-2023/WikiHow": gr.Dataset.load("ajibawa-2023/WikiHow"),
"ajibawa-2023/Code-74k-ShareGPT": gr.Dataset.load("ajibawa-2023/Code-74k-ShareGPT"),
"ajibawa-2023/Software-Architectural-Frameworks": gr.Dataset.load("ajibawa-2023/Software-Architectural-Frameworks"),
"ajibawa-2023/Python-Code-23k-ShareGPT": gr.Dataset.load("ajibawa-2023/Python-Code-23k-ShareGPT"),
"HuggingFaceFW/fineweb": gr.Dataset.load("HuggingFaceFW/fineweb"),
"kye/all-huggingface-python-code-2": gr.Dataset.load("kye/all-huggingface-python-code-2"),
"suvadityamuk/huggingface-transformers-code-dataset": gr.Dataset.load("suvadityamuk/huggingface-transformers-code-dataset")
}
# Define the interface
def generate_code(prompt, model, dataset, temperature, max_length):
model_instance = models[model]
dataset_instance = datasets[dataset]
output = model_instance.generate(prompt, dataset_instance, temperature, max_length)
return output
iface = gr.Interface(
fn=generate_code,
inputs=[
________gr.Textbox(label="Prompt"),
________gr.Dropdown(label="Model",_choices=list(models.keys())),
________gr.Dropdown(label="Dataset",_choices=list(datasets.keys())),
________gr.Slider(label="Temperature",_minimum=0.1,_maximum=1.0,_default=0.5),
________gr.Slider(label="Max_Length",_minimum=10,_maximum=1000,_default=200)
____],
outputs="text"
)
# Launch the interface
iface.launch()
|