diff --git a/apis.py b/apis.py
index 86f8e18..52899c5 100644
--- a/apis.py
+++ b/apis.py
@@ -97,9 +97,16 @@ menuLLM="""        <option value="meta-llama/Meta-Llama-3.1-70B-Instruct">meta-l
 
 @app.post("/uploadimg")
 def upload_image(image: UploadFile = File(...),type="factura"):
-    endfile=".jpg"
+    if image.headers['content-type']=='image/png':
+        endfile=".png"
+    elif image.headers['content-type']=='image/jpeg':
+        endfile=".jpg"
+    else:
+        return JSONResponse(content={
+            "message": f"The file should be png or jpg"
+        }, status_code=500)
+    
     t=time.time()
-    print(image.headers)
     try:
         # Create a temporary file to store the uploaded audio
         headfilename=uuid.uuid4()