jykoh commited on
Commit
3bd98a8
1 Parent(s): ccfd05b

Try docker

Browse files
Files changed (2) hide show
  1. Dockerfile +18 -0
  2. README.md +1 -2
Dockerfile ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime as base
2
+
3
+ RUN apt-get update && apt-get -y install git
4
+
5
+
6
+ ENV HOME=/exp/fromage
7
+
8
+
9
+
10
+ WORKDIR /exp/fromage
11
+ COPY ./requirements.txt ./requirements.txt
12
+ RUN python -m pip install -r ./requirements.txt
13
+ RUN python -m pip install gradio
14
+
15
+ COPY . .
16
+ RUN chmod -R a+rwX .
17
+
18
+ CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"]
README.md CHANGED
@@ -1,7 +1,6 @@
1
  title: FROMAGe
2
  emoji: 🧀
3
- sdk: gradio
4
- sdk_version: 3.17.0
5
  colorFrom: blue
6
  colorTo: indigo
7
  pinned: true
 
1
  title: FROMAGe
2
  emoji: 🧀
3
+ sdk: docker
 
4
  colorFrom: blue
5
  colorTo: indigo
6
  pinned: true