Update app.py

removes model_ name conflict

app.py

@@ -97,9 +97,8 @@ class PredictionRequest(BaseModel):
 
 class PredictionResponse(BaseModel):
     generated_text: str
-    model_used: str
+    selected_model: str  # Changed from model_used to avoid namespace conflict
 
-# Pythhon version check
 @app.get("/version")
 async def version():
     return {
@@ -107,26 +106,6 @@ async def version():
         "models_available": list(MODELS.keys())
     }
 
-# Environment check 
-@app.get("/debug/environment")
-async def get_environment_info():
-    import torch
-    import transformers
-    import platform
-    
-    return {
-        "python_version": platform.python_version(),
-        "torch_version": torch.__version__,
-        "torch_device": str(torch.device("cuda" if torch.cuda.is_available() else "cpu")),
-        "transformers_version": transformers.__version__,
-        "available_devices": {
-            "cuda_available": torch.cuda.is_available(),
-            "cuda_device_count": torch.cuda.device_count() if torch.cuda.is_available() else 0,
-            "mps_available": hasattr(torch.backends, "mps") and torch.backends.mps.is_available()
-        }
-    }
-
-# Health check
 @app.get("/health")
 async def health():
     # More comprehensive health check
@@ -144,7 +123,6 @@ async def health():
             "error": str(e)
         }
 
-# Main /predict endpoint
 @app.post("/predict", response_model=PredictionResponse)
 async def predict(request: PredictionRequest):
     try:
@@ -201,7 +179,7 @@ async def predict(request: PredictionRequest):
         
         return PredictionResponse(
             generated_text=result,
-            model_used=request.model
+            selected_model=request.model
         )
 
     except Exception as e: