Update Dockerfile

Dockerfile

@@ -8,6 +8,7 @@ RUN pip install --upgrade pip && pip install --no-cache-dir -r /app/requirements
 # Now copy model and code
 COPY Random_Forest_pipeline_for_hf.joblib /app/model.joblib
 COPY app.py /app/
+COPY main.py /app/
 
 EXPOSE 7860
 ENV PORT=7860