Geraldine commited on
Commit
398493c
·
verified ·
1 Parent(s): df7325b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,7 +58,7 @@ def describe_image(image_array):
58
  max_new_tokens=500,
59
  num_beams=1, # Disable beam search
60
  do_sample=False, # Disable sampling
61
- temperature=1.0 # Set temperature to 1.0
62
  )
63
  output_ids = [
64
  generated_ids[len(input_ids) :]
@@ -75,7 +75,7 @@ iface = gr.Interface(
75
  fn=describe_image,
76
  inputs=gr.Image(),
77
  outputs=[gr.Textbox(label="Description"), gr.JSON(label="Embeddings")],
78
- title="Image Description with SmolVLM Model",
79
  description="Upload an image to get a detailed description using the SmolVLM-500M-Instruct model."
80
  )
81
 
 
58
  max_new_tokens=500,
59
  num_beams=1, # Disable beam search
60
  do_sample=False, # Disable sampling
61
+ #temperature=1.0 # Set temperature to 1.0
62
  )
63
  output_ids = [
64
  generated_ids[len(input_ids) :]
 
75
  fn=describe_image,
76
  inputs=gr.Image(),
77
  outputs=[gr.Textbox(label="Description"), gr.JSON(label="Embeddings")],
78
+ title="Image Description with SmolVLM-500M-Instruct and Textual embeddings with all-MiniLM-L6-v2",
79
  description="Upload an image to get a detailed description using the SmolVLM-500M-Instruct model."
80
  )
81