Update main.py

main.py

@@ -1,4 +1,3 @@
-import os
 import io
 from flask import Flask, request, jsonify
 import base64
@@ -9,12 +8,9 @@ from keras.applications.xception import Xception #to get pre-trained model Xcept
 from keras.models import load_model
 from keras.preprocessing.sequence import pad_sequences
 
-os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
-
 app = Flask(__name__)
 
-# MAX_LENGTH = 34
-MAX_LENGTH = 38
+MAX_LENGTH = 34
 
 def extract_features(image_data, model):
     try:
@@ -65,16 +61,12 @@ def generate_desc(model, tokenizer, photo, max_length):
 def generate_caption():
     try:
         base64_image_data = request.form['image']
-        # return jsonify({'caption': base64_image_data}), 200
-        # Replace spaces with "+" characters to handle cases where "+" characters are missing
-        # base64_image_data = base64_image_data.replace(" ", "+")
         
         # Decode the Base64 string into binary image data
         image_data = base64.b64decode(base64_image_data)
 
         tokenizer = load(open("tokenizer.p","rb"))
-        # model = load_model('model_9.h5')
-        model = load_model('models/model_9.keras')
+        model = load_model('model_9.h5')
 
         xception_model = Xception(include_top=False, pooling="avg")
         photo = extract_features(image_data, xception_model)