finaly

Dockerfile

@@ -9,22 +9,35 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Create cache directory with proper permissions
+RUN mkdir -p /home/user/.cache/huggingface && \
+    chown -R user:user /home/user/.cache
+
+# Set environment variables
+ENV PYTHONPATH=/app
+ENV PORT=8000
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+ENV HF_HOME=/home/user/.cache/huggingface
+
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy the rest of the application
-COPY . .
+COPY --chown=user:user . .
 
 # Create necessary directories
-RUN mkdir -p src/web
+RUN mkdir -p src/web && \
+    chown -R user:user /app
 
-# Expose the port
-EXPOSE 8000
+# Switch to non-root user
+USER user
 
-# Set environment variables
-ENV PYTHONPATH=/app
-ENV PORT=8000
+# Expose the FastAPI port
+EXPOSE 8000
 
-# Command to run the application
+# Command to run the FastAPI application
 CMD ["uvicorn", "src.api.app:app", "--host", "0.0.0.0", "--port", "8000"]

README.md

@@ -4,9 +4,11 @@ emoji: 🚀
 colorFrom: red
 colorTo: red
 sdk: docker
-app_port: 8501
+app_port: 8000
 tags:
-  - streamlit
+  - fastapi
+  - medical
+  - ai
 pinned: false
 ---
 
@@ -14,6 +16,8 @@ pinned: false
 
 A synthetic medical text generator that creates realistic medical records using AI models. The application provides both a FastAPI backend and a Streamlit interface.
 
+> **Note**: This Hugging Face Space runs the FastAPI version with HTML interface. For the Streamlit version, please run locally using `streamlit run app.py`.
+
 ## Features
 
 - Generate various types of medical records:
@@ -26,8 +30,8 @@ A synthetic medical text generator that creates realistic medical records using
   - Hugging Face models (default)
   - Google Gemini (optional)
 - Two interfaces:
-  - FastAPI with HTML frontend
-  - Streamlit interface
+  - FastAPI with HTML frontend (Hugging Face Space)
+  - Streamlit interface (Local development)
 
 ## API Endpoints
 
@@ -76,10 +80,7 @@ A synthetic medical text generator that creates realistic medical records using
 
 ### Hugging Face Spaces Deployment
 
-1. Create a new Space on Hugging Face
-2. Choose "Docker" as the SDK
-3. Push this repository to your Space
-4. The application will be automatically deployed
+This Space runs the FastAPI version with HTML interface. The application is automatically deployed when you push to the repository.
 
 ## Environment Variables