Update app.py
Browse files
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
|
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 |
|