Update image.js

image.js

@@ -8,7 +8,7 @@ const { HfInference } = require("@huggingface/inference");
 require("dotenv").config()
 
 // Cache directory to store generated images
-const CACHE_DIR = `public/image_out`
+const CACHE_DIR = `/home/root/code/public/image_out`
 
 
 if (!fs.existsSync(CACHE_DIR)) {
@@ -70,6 +70,7 @@ const generateImage = async(requestPayload, res, responseFormat) => {
   console.log(cachePath)
   // Check if the image already exists in the cache
   if (fs.existsSync(cachePath)) {
+    console.log("Found in cache! "+cacheKey)
     return res.sendFile(cachePath);
   }
 
@@ -114,8 +115,9 @@ const generateImage = async(requestPayload, res, responseFormat) => {
       res.json({"image_url": "https://defactofficial-mmapi.hf.space/image_out/"+ cachePath.split("/").at(-1)})
       res.end()
     } else {
-      res.setHeader('Content-Type', 'image/png');
+      res.setHeader('Content-Type', 'application/octet-stream');
       res.sendFile(cachePath);
+      res.end()
     }
   } catch (error) {
     res.status(500).send(`Error: Failed to generate image - ${error.message}`);