initial commit

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . .
+
+CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "7860"]

docker-compose.yaml

@@ -0,0 +1,12 @@
+services:
+  server:
+    build:
+      context: .
+    ports:
+      - 7860:7860
+    environment:
+      - foo=bar
+    develop:
+      watch:
+        - action: rebuild
+          path: .

requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.111.*
+requests==2.*
+uvicorn[standard]==0.30.*
+torch

src/main.py

@@ -0,0 +1,33 @@
+import os
+import requests
+import torch
+
+#from typing import Optional
+
+from fastapi import FastAPI, Header, HTTPException, BackgroundTasks
+from fastapi.responses import FileResponse
+#from huggingface_hub.hf_api import HfApi
+
+#from .models import config, WebhookPayload
+
+#WEBHOOK_SECRET = os.getenv("WEBHOOK_SECRET")
+#HF_ACCESS_TOKEN = os.getenv("HF_ACCESS_TOKEN")
+#AUTOTRAIN_API_URL = "https://api.autotrain.huggingface.co"
+#AUTOTRAIN_UI_URL = "https://ui.autotrain.huggingface.co"
+
+
+app = FastAPI()
+
+@app.get("/")
+async def home():
+
+    gpu = 'GPU not available'
+    if torch.cuda.is_available():
+        gpu = 'GPU is available'
+        print("GPU is available")
+    else:
+        print("GPU is not available")
+
+    print('hello world')
+    print(os.getenv("foo"))
+    return {'success': True, 'response': 'hello world 3', 'gpu': gpu}