Update app.py

app.py

@@ -1,5 +1,24 @@
-# requirements: transformers, torch, peft, accelerate, bitsandbytes, gradio, sentencepiece, protobuf
+import subprocess
+import sys
+import os
 
+# Install required packages directly in the app
+def install(package):
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+
+print("Installing required packages...")
+install("transformers==4.56.0")
+install("torch==2.3.0")
+install("peft==0.12.0")
+install("accelerate==0.28.0")
+install("bitsandbytes==0.43.0")
+install("sentencepiece")
+install("protobuf")
+install("gradio==5.27.0")
+
+print("All packages installed. Loading model...")
+
+# Now import after installation
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
@@ -13,14 +32,13 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype="auto",
     device_map="auto",
     trust_remote_code=True,
-    use_cache=False  # Important for generation
+    use_cache=False
 )
 
 # Load your fine-tuned adapter
 model = PeftModel.from_pretrained(model, "KlemGunn0519/Mighty_Mountain_Ski_Resort")
 
 def respond(message, history):
-    # Format prompt as in training
     prompt = f"### Instruction\n{message}\n\n### Response"
     inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     
@@ -33,23 +51,18 @@ def respond(message, history):
         pad_token_id=tokenizer.eos_token_id
     )
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    # Extract only the assistant's response (after "### Response")
-    try:
-        return response.split("### Response")[-1].strip()
-    except:
-        return response
+    return response.split("### Response")[-1].strip()
 
-# Clean Gradio Chat Interface
+# Create Gradio interface
 demo = gr.ChatInterface(
     fn=respond,
     title="🎿 Mighty Mountain Ski Resort Assistant",
-    description="Ask about lift tickets, trail conditions, hours, rentals, and more!",
+    description="Ask about lift tickets, trail conditions, hours, and more!",
     examples=[
         "What are daily lift ticket prices?",
-        "Do you have night skiing?",
         "What's the best beginner slope?",
-        "How do I buy a season pass?"
+        "Do you offer ski rentals?",
+        "When does the resort open for the season?"
     ],
     theme="soft",
     retry_btn=None,