Deploy recipe generator to HF Spaces

.gitignore

@@ -1,3 +1,5 @@
 __pycache__/
 *.pyc
-.streamlit/
+.streamlit/
+venv/
+*.env

Dockerfile

@@ -7,4 +7,6 @@ RUN pip install -r requirements.txt
 
 COPY . .
 
-CMD streamlit run app.py --server.port=7860
+EXPOSE 7860
+
+CMD streamlit run app.py --server.port=7860 --server.address=0.0.0.0

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 from transformers import pipeline
 
@@ -9,7 +10,8 @@ st.write("Enter ingredients and get a recipe suggestion!")
 # Load the model (runs once, cached)
 @st.cache_resource
 def load_model():
-    return pipeline('text-generation', model='mistralai/Mistral-7B-Instruct-v0.1')
+    token = os.environ.get("HF_TOKEN")
+    return pipeline('text-generation', model='mistralai/Mistral-7B-Instruct-v0.1', token=token)
 
 model = load_model()
 

requirements.txt

@@ -1,3 +1,4 @@
 streamlit>=1.28.0
 transformers>=4.35.0
-torch>=2.6.0
+torch>=2.6.0
+accelerate>=0.26.0