Update app.py

app.py

@@ -1,40 +1,20 @@
-import gradio as gr
-from onnxruntime_genai import ChatSession, GenerationConfig, ORTModel
-
-# Path to the downloaded ONNX model
-MODEL_DIR = "model/cpu_and_mobile/cpu-int4-rtn-block-32-acc-level-4"
-
-# Load the model
-model = ORTModel(MODEL_DIR, execution_provider="cpu")
-session = ChatSession(model)
-
-# Chat function
-def chat_with_phi4(message, history):
-    history = history or []
-    for past_msg in history:
-        session.history.append((past_msg[0], past_msg[1]))
-
-    reply = session.chat(message, config=GenerationConfig(max_new_tokens=300))
-    history.append((message, reply))
-    return history, history
-
-# Hugging Face logo
-HF_LOGO = "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"
-
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Image(value=HF_LOGO, width=120, show_label=False, show_download_button=False)
-    gr.Markdown("### Chat with Microsoft Phi-4 Mini Instruct (ONNX)")
-
-    chatbot = gr.Chatbot()
-    user_input = gr.Textbox(label="Your message")
-    state = gr.State([])
-
-    send_btn = gr.Button("Send")
-    clear_btn = gr.Button("Clear")
-
-    send_btn.click(chat_with_phi4, [user_input, state], [chatbot, state])
-    user_input.submit(chat_with_phi4, [user_input, state], [chatbot, state])
-    clear_btn.click(lambda: ([], []), outputs=[chatbot, state])
-
-demo.launch()
+# Core Gradio UI library
+gradio>=4.0.0,<5.0.0
+
+# ONNX Runtime GenAI for inference (CPU version)
+# Use --pre as it's often in pre-release
+# NOTE: If targeting a GPU Space, change this to onnxruntime-genai-cuda
+# and update EXECUTION_PROVIDER in app.py to "cuda"
+onnxruntime-genai --pre
+
+# Hugging Face Hub for downloading models/files
+huggingface_hub>=0.20.0
+
+# ONNX Runtime itself (will be installed as a dependency of onnxruntime-genai,
+# but specifying can sometimes help resolve version conflicts if needed)
+# onnxruntime>=1.17.0 # Generally not needed to list explicitly
+
+# Git LFS is needed by huggingface_hub to download large model files.
+# It needs to be installed on the Space environment, which is usually handled
+# by the Hugging Face Spaces infrastructure if not using Docker.
+# If you encounter download issues, ensure git-lfs is available.