File size: 620 Bytes
fde41fa
20fd975
fde41fa
cabdd14
fde41fa
cabdd14
 
 
 
20fd975
fde41fa
20fd975
fde41fa
 
 
 
 
 
20fd975
 
fde41fa
20fd975
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
fine_tuned_model = "ashioyajotham/results"

def generate_code(input_text):
    inputs = tokenizer.encode(input_text, return_tensors='pt')
    # Use your fine-tuned model to generate code here
    generated_code = fine_tuned_model.generate(input_text, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
    completion = tokenizer.decode(generated_code[0])

    return completion

import gradio as gr

iface = gr.Interface(
    fn=generate_code,
    inputs="text",
    outputs="text",
    title="Code Generation App",
    description="Generate code from text input."
)

iface.launch()