Init commit

.gitignore

@@ -0,0 +1 @@
+.idea

Dockerfile

@@ -0,0 +1,28 @@
+# Start from the TGI base image
+FROM [Your TGI Base Image] as base
+
+# Install JupyterLab
+# Note: Ensure Python is installed in the base image. If not, you will need to install it.
+RUN pip install jupyterlab
+
+# Copy any necessary files (if needed)
+# COPY your-files /your-destination
+
+# AWS Sagemaker compatible image
+# Assuming this part remains the same from your original Dockerfile
+FROM base as sagemaker
+
+COPY sagemaker-entrypoint.sh entrypoint.sh
+RUN chmod +x entrypoint.sh
+
+ENTRYPOINT ["./entrypoint.sh"]
+
+# Final image
+FROM base
+
+# Add JupyterLab entrypoint
+# Note: You can customize the command to suit your needs
+ENTRYPOINT ["jupyter", "lab", "--ip=0.0.0.0", "--allow-root", "--NotebookApp.token='' --port 7860"]
+
+# Optional: Set CMD to launch TGI or any other command
+CMD ["text-generation-launcher", "--json-output"]

README.md

@@ -1,5 +1,5 @@
 ---
-title: Jais Tgi Benchmark
+title: Jais TGI Benchmark
 emoji: 👁
 colorFrom: yellow
 colorTo: gray