Create app.py

app.py

@@ -65,4 +65,24 @@ def process_image():
 
 if __name__ == '__main__':
     # Run the app
-    app.run(host='0.0.0.0', port=5000, debug=True)
+    app.run(host='0.0.0.0', port=5000, debug=True)
+    import os
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
+import tensorflow as tf
+print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
+import os
+import tensorflow as tf
+
+# Set environment variable
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
+
+# Check GPU availability
+print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
+
+# Your main code goes here
+def main():
+    # Your application code
+    pass
+
+if __name__ == "__main__":
+    main()