gradio

Dockerfile

@@ -24,5 +24,7 @@ RUN chmod +x /usr/local/bin/sft
 # Optional: set entrypoint to sft
 ENTRYPOINT ["sft"]
 
-# Default command to run the script
-CMD ["python", "main.py"]
+# Launch the application
+EXPOSE 7860
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,6 @@
+import gradio as gr
+
+def greet(name):
+    return f"Hello {name}!"
+
+iface = gr.Interface(fn=greet, inputs="text", outputs="text").launch()

main.py

@@ -1 +0,0 @@
-print("This is main!")

requirements.txt

@@ -1 +1,2 @@
-git+https://github.com/huggingface/trl.git@main#egg=trl
+git+https://github.com/huggingface/trl.git@main#egg=trl
+gradio