Update app.py

app.py

@@ -3,6 +3,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import torch
 import spaces
 
+MODEL_PATH = "benhaotang/mistral-small-physics-finetuned-bnb-4bit"
+MODEL_URL = f"https://huggingface.co/{MODEL_PATH}"
+
 def load_model():
     bnb_config = BitsAndBytesConfig(
         load_in_8bit=False,
@@ -40,7 +43,9 @@ demo = gr.Interface(
     ],
     outputs=gr.Textbox(label="Response", lines=10),
     title="Physics AI Assistant",
-    description="Ask questions about physics concepts, and I'll provide detailed explanations.",
+    description=f"""Ask questions about physics concepts, and I'll provide detailed explanations.
+    
+Model: [benhaotang/mistral-small-physics-finetuned-bnb-4bit]({MODEL_URL})""",
     examples=[
         ["Give me a short introduction to renormalization group(RG) flow in physics?"],
         ["What is quantum entanglement?"],