hermanhelf commited on
Commit
6d2ed05
1 Parent(s): 24edd57

remove the existence of the token slider

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -26,8 +26,10 @@ processor = PaliGemmaProcessor.from_pretrained(model_id)
26
  def infer(
27
  image: PIL.Image.Image,
28
  text: str,
29
- max_new_tokens: int
30
  ) -> str:
 
 
31
  inputs = processor(text=text, images=image, return_tensors="pt").to(device)
32
  with torch.inference_mode():
33
  generated_ids = model.generate(
@@ -114,8 +116,8 @@ with gr.Blocks(css="style.css") as demo:
114
  ]
115
  chat_btn.click(
116
  fn=infer,
117
- # inputs=chat_inputs,
118
- inputs=chat_inputs + [tokens],
119
  outputs=chat_outputs,
120
  )
121
 
 
26
  def infer(
27
  image: PIL.Image.Image,
28
  text: str,
29
+ # max_new_tokens: int
30
  ) -> str:
31
+ # set this to hide the slider for a bit.
32
+ max_new_tokens = 20
33
  inputs = processor(text=text, images=image, return_tensors="pt").to(device)
34
  with torch.inference_mode():
35
  generated_ids = model.generate(
 
116
  ]
117
  chat_btn.click(
118
  fn=infer,
119
+ inputs=chat_inputs,
120
+ # inputs=chat_inputs + [tokens],
121
  outputs=chat_outputs,
122
  )
123