Create app.py

app.py

@@ -0,0 +1,48 @@
+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()
+