anilguleroglu commited on
Commit
7552761
·
verified ·
1 Parent(s): 09283df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,7 +20,7 @@ model, tokenizer = load_model_and_tokenizer()
20
  # ------------------------------------------------
21
  # Metin üretim fonksiyonu
22
  # ------------------------------------------------
23
- def generate_text(prompt, max_length=512):
24
  inputs = tokenizer(prompt, return_tensors="pt")
25
 
26
  outputs = model.generate(
@@ -33,7 +33,7 @@ def generate_text(prompt, max_length=512):
33
  do_sample=True,
34
  top_k=10,
35
  top_p=0.9,
36
- temperature=0.1,
37
  repetition_penalty=1.3
38
  )
39
 
@@ -44,7 +44,7 @@ def generate_text(prompt, max_length=512):
44
  print(response)
45
  # Anahtar kelimeden sonrasını çıkarma
46
  if keyword in response:
47
- extracted_content = response.split(keyword, 1)[1].strip()
48
  return extracted_content
49
  else:
50
  response
 
20
  # ------------------------------------------------
21
  # Metin üretim fonksiyonu
22
  # ------------------------------------------------
23
+ def generate_text(prompt, max_length=1024):
24
  inputs = tokenizer(prompt, return_tensors="pt")
25
 
26
  outputs = model.generate(
 
33
  do_sample=True,
34
  top_k=10,
35
  top_p=0.9,
36
+ temperature=0.3,
37
  repetition_penalty=1.3
38
  )
39
 
 
44
  print(response)
45
  # Anahtar kelimeden sonrasını çıkarma
46
  if keyword in response:
47
+ extracted_content = response.split(keyword, 1)[1].strip().replace("<|eot_id|>")
48
  return extracted_content
49
  else:
50
  response