AdrienB134 commited on
Commit
e56b146
1 Parent(s): b8734cc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
28
 
29
  @spaces.GPU
30
  def model_inference(
31
- images, text, assistant_prefix= None, decoding_strategy = "Greedy", temperature= 0.4, max_new_tokens=512,
32
  repetition_penalty=1.2, top_p=0.8
33
  ):
34
  ## Load idefics
 
28
 
29
  @spaces.GPU
30
  def model_inference(
31
+ images, text, assistant_prefix= "Répond en faisant de belles phrases", decoding_strategy = "Greedy", temperature= 0.4, max_new_tokens=512,
32
  repetition_penalty=1.2, top_p=0.8
33
  ):
34
  ## Load idefics