commit

main.py

@@ -12,41 +12,19 @@ from PIL import Image
 from typing import Any, Dict
 import os
 import pkgutil
-print('starlette', __import__('starlette').__version__)
+
 
 from huggingface_hub import hf_hub_download
 from typing import Any, Dict
 
 import tensorflow as tf
-print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
-if tf.config.list_physical_devices('GPU'):
-    print(tf.config.list_physical_devices('GPU')[0].device_type, tf.config.list_physical_devices('GPU')[0].name )
+
+
 
 from tensorflow import keras
-TF_AVAILABLE = True
-print("tensorflow version: ", tf.__version__)
-print("keras version: ", keras.__version__)
-
-
-# starlette is a FastAPI dependency; import if available
-try:
-	import starlette
-except Exception:
-	starlette = None
-
-# TensorFlow can be large or absent in some envs; guard the import so
-# importing this module doesn't crash tests or other tooling.
-try:
-	import tensorflow as tf
-	from tensorflow import keras
-	TF_AVAILABLE = True
-	print("tensorflow version: ", tf.__version__)
-	print("keras version: ", keras.__version__)
-except Exception:
-	tf = None
-	keras = None
-	TF_AVAILABLE = False
-	print("tensorflow not available")
+
+
+
 
 
 app = FastAPI(title="1.3 - AI Model Deployment")
@@ -112,26 +90,15 @@ def versions() -> Dict[str, Any]:
     """Return key package versions and whether TensorFlow is available."""
     return {
         "fastapi": fastapi.__version__,
-        "starlette": getattr(starlette, "__version__", None),
-        "tensorflow_available": TF_AVAILABLE,
-        "tensorflow_version": getattr(tf, "__version__", None),
+
     }
 
 
 @app.get("/predict")
 def predict_stub() -> Dict[str, Any]:
-    """A tiny predict stub that demonstrates how to expose model inference.
-
-    If TensorFlow isn't available or no model is loaded this returns a helpful
-    message.
-    """
-    if not TF_AVAILABLE or model is None:
-        return {
-            "prediction": "N/A",
-            "info": "TensorFlow not available or model not loaded.",
-        }
+
     # This is a stub, so we're not doing a real prediction
-    return {"prediction": "stub, we're not doing a real prediction", "model_path": model_path}
+    return {"prediction": "stub, we're not doing a real prediction"}