Update tensorflow

app.py

@@ -8,7 +8,7 @@ import urllib.request
 import numpy as np
 from flask import Flask, jsonify, render_template, request
 from PIL import Image
-
+import tensorflow as tf
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -126,11 +126,16 @@ def load_pytorch():
 
 def load_tensorflow():
     global _tf_model
-    if _tf_model is None:
-        import tensorflow as tf
-        _tf_model = tf.keras.models.load_model("parfait_model.keras")
-    return _tf_model
+    if _tf_model is not None:
+        return _tf_model
 
+    import tensorflow as tf
+
+    model = build_cnn_tf(num_classes=6, input_shape=(228, 228, 3))
+    model.load_weights("parfait_model.keras")
+    _tf_model = model
+    return _tf_model
+    
 
 def read_input_image():
     if "image" in request.files and request.files["image"].filename: