Update app.py

app.py

@@ -10,6 +10,7 @@ import io
 import logging
 import time
 import os
+import tempfile
 from urllib.parse import urlparse
 
 # Set up logging
@@ -139,10 +140,24 @@ def load_model(model_name):
         if error:
             return None, error
         
-        # Try to load the model
+        # Try to load the model using temporary file
         try:
-            model_bytes.seek(0)
-            model = tf.keras.models.load_model(model_bytes, compile=False)
+            import tempfile
+            
+            # Create temporary file
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.keras') as tmp_file:
+                model_bytes.seek(0)
+                tmp_file.write(model_bytes.read())
+                tmp_file_path = tmp_file.name
+            
+            # Load model from temporary file
+            model = tf.keras.models.load_model(tmp_file_path, compile=False)
+            
+            # Clean up temporary file
+            try:
+                os.unlink(tmp_file_path)
+            except:
+                pass  # Ignore cleanup errors
             
             # Cache the model
             model_cache[model_name] = model
@@ -151,6 +166,12 @@ def load_model(model_name):
             
         except Exception as load_error:
             logger.error(f"Model loading error for {model_name}: {str(load_error)}")
+            # Try to cleanup temp file if it exists
+            try:
+                if 'tmp_file_path' in locals():
+                    os.unlink(tmp_file_path)
+            except:
+                pass
             return None, f"Model loading failed: {str(load_error)}"
             
     except Exception as e: