Halleq commited on
Commit
36c5d9a
·
verified ·
1 Parent(s): 7551575

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")
10
  # دالة التوليد
11
  def generate_code(prompt):
12
  inputs = tokenizer(prompt, return_tensors="pt")
13
- outputs = model.generate(**inputs, max_length=128, num_return_sequences=1)
14
  code = tokenizer.decode(outputs[0], skip_special_tokens=True)
15
  return code
16
 
 
10
  # دالة التوليد
11
  def generate_code(prompt):
12
  inputs = tokenizer(prompt, return_tensors="pt")
13
+ outputs = model.generate(**inputs, max_new_tokens=50)
14
  code = tokenizer.decode(outputs[0], skip_special_tokens=True)
15
  return code
16