Inan Ince commited on
Commit
eb58738
1 Parent(s): a30219c

Add application file17

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -8,13 +8,15 @@ 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=300, # Çıktının uzunluğunu artır
17
- num_beams=5, # Daha kaliteli sonuçlar için beam search
18
  early_stopping=True
19
  )
20
  # Tokenize edilmiş çıktıyı kod olarak döndür
@@ -23,6 +25,9 @@ def generate_java_code(prompt):
23
  # Post-processing: Eksik yapıları düzelt
24
  if "public static void main" not in code:
25
  code = f"public class Main {{\n{code}\n}}"
 
 
 
26
  return code
27
 
28
  # Gradio UI tasarımı
 
8
 
9
  # Kod üretme fonksiyonu
10
  def generate_java_code(prompt):
11
+ # Girdi metnini modele uygun bir şekilde biçimlendir
12
+ prompt = f"Generate a complete Java program:\n{prompt}"
13
+
14
+ # Girdiyi tokenize et ve modele gönder
15
  inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
 
16
  outputs = model.generate(
17
  inputs["input_ids"],
18
  max_length=300, # Çıktının uzunluğunu artır
19
+ num_beams=5, # Beam search ile kaliteyi artır
20
  early_stopping=True
21
  )
22
  # Tokenize edilmiş çıktıyı kod olarak döndür
 
25
  # Post-processing: Eksik yapıları düzelt
26
  if "public static void main" not in code:
27
  code = f"public class Main {{\n{code}\n}}"
28
+ if code.strip().endswith("{"):
29
+ code += "\n // Complete the logic here\n}"
30
+
31
  return code
32
 
33
  # Gradio UI tasarımı