CodeAlpaca / app.py
sahil2801's picture
output format
840019f
raw history blame
No virus
644 Bytes
import gradio as gr
import banana_dev as banana
import os
api_key = os.getenv("API_KEY")
model_key = os.getenv("MODEL_KEY")
title = "CodeAlpaca"
description = "Instruction following model for code generation."
examples = ["Write a python function to sort a list."]
def call_model(prompt):
model_parameters = {
"prompt": prompt,
"max_length":1000
}
out = banana.run(api_key, model_key, model_parameters)
return out["modelOutputs"][0]["output"]
iface = gr.Interface(fn=call_model, inputs="text", outputs=gr.code(label="Output",lines=20),examples=examples,description=description,title=title)
iface.launch()