Update app.py
Browse files
app.py
CHANGED
@@ -39,11 +39,14 @@ def generate_embeddings(text):
|
|
39 |
embeddings = model.encode(sentences)
|
40 |
return embeddings
|
41 |
|
42 |
-
def describe_image(
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
|
|
|
|
|
|
47 |
|
48 |
messages = [
|
49 |
{
|
|
|
39 |
embeddings = model.encode(sentences)
|
40 |
return embeddings
|
41 |
|
42 |
+
def describe_image(image_array):
|
43 |
+
if image_array is None:
|
44 |
+
raise ValueError("No image provided. Please upload an image before submitting.")
|
45 |
+
# Convert numpy array to PIL Image
|
46 |
+
image = Image.fromarray(np.uint8(image_array))
|
47 |
+
buffered = io.BytesIO()
|
48 |
+
image.save(buffered, format="PNG") # Change format as needed
|
49 |
+
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
50 |
|
51 |
messages = [
|
52 |
{
|