Inan Ince commited on
Commit
5f090e3
1 Parent(s): ce236d9

Add application file15

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -1,22 +1,30 @@
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
- # CodeT5 modelini yükleyin
5
  model_name = "Salesforce/codet5-base"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
8
 
9
- # Java kodu üretme fonksiyonu
10
  def generate_java_code(prompt):
 
11
  inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
12
- outputs = model.generate(inputs["input_ids"], max_length=150, num_beams=4, early_stopping=True)
 
 
 
 
 
 
 
13
  code = tokenizer.decode(outputs[0], skip_special_tokens=True)
14
  return code
15
 
16
- # Gradio UI
17
  with gr.Blocks() as demo:
18
- gr.Markdown("<h1 style='text-align: center;'>Java Kod Üretici</h1>")
19
- prompt = gr.Textbox(label="Java kodu için doğal dil açıklaması", placeholder="Örnek: 2 sayıdan büyüğünü getiren bir Java kodu yaz.")
20
  output_code = gr.Textbox(label="Üretilen Java Kodu", lines=10)
21
  btn = gr.Button("Kod Üret")
22
  btn.click(generate_java_code, inputs=prompt, outputs=output_code)
 
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
+ # CodeT5 modelini yükle
5
  model_name = "Salesforce/codet5-base"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
8
 
9
+ # Kod üretme fonksiyonu
10
  def generate_java_code(prompt):
11
+ # Girdi metnini tokenize et
12
  inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
13
+ # Modelden kod üret
14
+ outputs = model.generate(
15
+ inputs["input_ids"],
16
+ max_length=150, # Çıktı kodunun maksimum uzunluğu
17
+ num_beams=4, # Beam search ile daha kaliteli sonuçlar
18
+ early_stopping=True
19
+ )
20
+ # Tokenize edilmiş çıktıyı kod olarak döndür
21
  code = tokenizer.decode(outputs[0], skip_special_tokens=True)
22
  return code
23
 
24
+ # Gradio UI tasarımı
25
  with gr.Blocks() as demo:
26
+ gr.Markdown("<h1 style='text-align: center;'>Java Kod Üretici (CodeT5)</h1>")
27
+ prompt = gr.Textbox(label="Doğal Dil Girdisi", placeholder="Örnek: Write a Java program to find the larger of two numbers.")
28
  output_code = gr.Textbox(label="Üretilen Java Kodu", lines=10)
29
  btn = gr.Button("Kod Üret")
30
  btn.click(generate_java_code, inputs=prompt, outputs=output_code)