abaryan commited on
Commit
9e0ed80
·
verified ·
1 Parent(s): 1a8cd16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -101,10 +101,10 @@ def predict_skin_lesion(image):
101
  generated_ids = model.generate(
102
  **inputs,
103
  max_new_tokens=200,
104
- temperature=0.3,
105
- top_p=0.8,
106
  # repetition_penalty=1.1,
107
- do_sample=True,
108
  pad_token_id=processor.tokenizer.eos_token_id
109
  )
110
 
 
101
  generated_ids = model.generate(
102
  **inputs,
103
  max_new_tokens=200,
104
+ temperature=0.6,
105
+ top_p=0.9,
106
  # repetition_penalty=1.1,
107
+ # do_sample=True,
108
  pad_token_id=processor.tokenizer.eos_token_id
109
  )
110