Upload 3 files

Dockerfile

@@ -1,21 +1,10 @@
-FROM python:3.9-slim
+FROM python:3.10-slim
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-COPY requirements.txt ./
-COPY src/ ./src/
+COPY . .
 
-RUN pip3 install -r requirements.txt
-
-EXPOSE 8501
-
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
-
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.enableCORS=false"]

README.md

@@ -1,19 +1,5 @@
----
-title: MaaGpt
-emoji: 🚀
-colorFrom: red
-colorTo: red
-sdk: docker
-app_port: 8501
-tags:
-- streamlit
-pinned: false
-short_description: Recipes with a Mother’s Touch
----
+# MaaGPT 👵🍲
 
-# Welcome to Streamlit!
+MaaGPT is a Streamlit app powered by Phi-2 LLM. It acts like a sweet Indian mother who gives advice about food with love and sass.
 
-Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
-
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+🧡 Try asking for pizza or Maggi and see how Maa reacts.

app.py

@@ -0,0 +1,29 @@
+import streamlit as st
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+
+@st.cache_resource
+def load_model():
+    model_id = "microsoft/phi-2"
+    tokenizer = AutoTokenizer.from_pretrained(model_id)
+    model = AutoModelForCausalLM.from_pretrained(model_id)
+    return pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+generator = load_model()
+
+def maa_prompt(user_input):
+    return f"""
+You are a caring Indian mother. If your child asks for junk food (like chips, pizza, maggi, etc.), respond with love and a little scolding.
+
+Child: {user_input}
+Mom:"""
+
+st.set_page_config(page_title="MaaGPT", page_icon="👵")
+st.title("👵 MaaGPT: Your Indian Mom AI")
+
+user_input = st.text_input("What do you feel like eating, beta? 😋")
+
+if user_input:
+    prompt = maa_prompt(user_input)
+    result = generator(prompt, max_new_tokens=100, do_sample=True)[0]["generated_text"]
+    response = result.split("Mom:")[-1].strip()
+    st.markdown(f"### 👵 Maa says:\n\n{response}")