Update Dockerfile

Dockerfile

@@ -1,8 +1,21 @@
-FROM python:3.9
-WORKDIR /app
-    
+FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
+
+# Install system packages
+RUN apt-get update && apt-get install -y \
+    python3 python3-pip git && \
+    ln -s /usr/bin/python3 /usr/bin/python
+
+# Install Python libraries
 COPY requirements.txt .
+RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
-    
-COPY . .
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Copy the app
+WORKDIR /app
+COPY app.py .
+
+# Expose the port required by Hugging Face Spaces or your own server
+EXPOSE 7860
+
+# Run FastAPI app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]