Update main.py

main.py

@@ -5,6 +5,7 @@ import json
 import io
 import random
 import string
+from mimetypes import guess_type
 
 def get_answer(image: io.BytesIO, question: str):
     filename = image.name
@@ -14,12 +15,13 @@ def get_answer(image: io.BytesIO, question: str):
     image.seek(0, 2)
     file_size = image.tell()
     image.seek(0)
+    image_mime_type, _ = guess_type(image.name)
     data = [
             {
                 "meta": {
                     "_type": "gradio.FileData"
                 },
-                "mime_type": "image/jpeg",
+                "mime_type": image_mime_type,
                 "orig_name": filename,
                 "path": filelink.split("=", 1)[1],
                 "size": file_size,