Hugging Face's logo

Spaces:
panduwana
/
interview-ai-detector
Running

App Files Community
bearking58 commited on
Commit
9c34cdf
·
1 Parent(s): 09418f5

fix: try smaller docker image + new weights

Browse files
Files changed (4) hide show
  1. core-model-prediction/Dockerfile +1 -1
  2. core-model-prediction/models/random_forest.joblib +0 -0
  3. core-model-prediction/requirements.txt +3 -3
  4. core-model-prediction/scalers/rf_scaler.joblib +0 -0
core-model-prediction/Dockerfile CHANGED
@@ -1,5 +1,5 @@
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
 
1
  # Use an official Python runtime as a base image
2
+ FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
3
 
4
  # Set the working directory in the container
5
  WORKDIR /app
core-model-prediction/models/random_forest.joblib CHANGED
Binary files a/core-model-prediction/models/random_forest.joblib and b/core-model-prediction/models/random_forest.joblib differ
 
core-model-prediction/requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- nltk
2
- vaderSentiment
3
  pandas
4
- textstat
5
  scikit-learn==1.2.2
6
  transformers==4.38.2
7
  fastapi
 
1
+ nltk==3.2.4
2
+ vaderSentiment==3.3.2
3
  pandas
4
+ textstat==0.7.3
5
  scikit-learn==1.2.2
6
  transformers==4.38.2
7
  fastapi
core-model-prediction/scalers/rf_scaler.joblib CHANGED
Binary files a/core-model-prediction/scalers/rf_scaler.joblib and b/core-model-prediction/scalers/rf_scaler.joblib differ