Second commit for deployment

.env

@@ -0,0 +1,7 @@
+DEBUG=False
+MODEL_PATH=./models/model.pt
+TEXT_MODEL=bert-base-uncased
+VISION_MODEL=google/vit-base-patch16-384
+HF_MODEL_REPO=dixisouls/VQA
+HF_MODEL_FILENAME=model.pt
+UPLOAD_DIR=./uploads

Dockerfile

@@ -0,0 +1,25 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+# Copy requirements first for better caching
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application
+COPY app/ ./app/
+COPY models/ ./models/
+COPY .env .
+
+# Create required directories
+RUN mkdir -p uploads models
+
+# Set environment variables
+ENV PYTHONPATH=/app
+ENV PORT=7860
+
+# Expose the port
+EXPOSE 7860
+
+# Start the application
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/config.py

@@ -35,6 +35,9 @@ class Settings(BaseSettings):
     # CORS settings
     ALLOW_ORIGINS: list[str] = ["*"]
     
+    # Hugging Face Spaces specific settings
+    PORT: int = int(os.getenv("PORT", "7860"))
+    
     class Config:
         env_file = ".env"
         case_sensitive = True

app/main.py

@@ -10,6 +10,7 @@ from contextlib import asynccontextmanager
 
 from app.routers import vqa
 from app.services.model_service import ModelService
+from app.config import settings
 
 # Configure logging
 logging.basicConfig(
@@ -38,10 +39,10 @@ app = FastAPI(
     lifespan=lifespan
 )
 
-# Add CORS middleware
+# Add CORS middleware with more permissive settings for Hugging Face Spaces
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allow all origins in development
+    allow_origins=["*"],  # Allow all origins
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -63,6 +64,19 @@ async def health_check():
         raise HTTPException(status_code=503, detail="Model not loaded")
     return {"status": "healthy", "model_loaded": True}
 
+# Root endpoint for Hugging Face Spaces
+@app.get("/")
+async def root():
+    """Root endpoint returning basic info about the API"""
+    return {
+        "name": "VizWiz Visual Question Answering API",
+        "description": "API for visual question answering on images",
+        "documentation": "/docs",
+        "health_check": "/health",
+        "version": "1.0.0"
+    }
+
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run("app.main:app", host="0.0.0.0", port=8000, reload=True)
+    port = settings.PORT
+    uvicorn.run("app.main:app", host="0.0.0.0", port=port, reload=False)