Update app.py

app.py

@@ -10,13 +10,17 @@ from PIL import Image
 import google.generativeai as genai
 from typing import Dict, List, Tuple, Optional
 
+import os
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+
 # Configure Gemini AI
 # You'll need to set your API key: export GOOGLE_API_KEY="your_api_key_here"
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 gemini_model = genai.GenerativeModel('gemini-1.5-flash')
 
 # Load the pre-trained ResNet model
-@gr.utils.cache
+
 def load_heartbeat_model():
     try:
         model = tf.keras.models.load_model('Heart_ResNet.h5')