Spaces:
Running
Running
Commit
·
f907a83
1
Parent(s):
0beb569
fix: change dockerfile to pytorch default
Browse files
core-model-prediction/Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Use an official Python runtime as a base image
|
| 2 |
-
FROM
|
| 3 |
|
| 4 |
# Set the working directory in the container
|
| 5 |
WORKDIR /app
|
|
|
|
| 1 |
# Use an official Python runtime as a base image
|
| 2 |
+
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-devel
|
| 3 |
|
| 4 |
# Set the working directory in the container
|
| 5 |
WORKDIR /app
|
core-model-prediction/requirements.txt
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
torch
|
| 2 |
nltk
|
| 3 |
vaderSentiment
|
| 4 |
pandas
|
|
|
|
|
|
|
| 1 |
nltk
|
| 2 |
vaderSentiment
|
| 3 |
pandas
|