updates to start

.gitignore

@@ -1,4 +1,5 @@
 dschatbot/
+venv/
 .DS_Store
 .env
 

app.py

@@ -1,17 +1,14 @@
-from ldclient import Context
 import os
 import logging
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import uuid
 import time
 import ldclient
 
 from ldclient.config import Config
-
-ldclient.set_config(Config(os.environ.get('sdkKEY')))
-client = ldclient.get()
+from ldclient import Context
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 def capture_logs(log_body, log_file, uuid_label):
@@ -41,6 +38,10 @@ def capture_logs(log_body, log_file, uuid_label):
 print("CUDA available: ", torch.cuda.is_available())
 print("MPS available: ", torch.backends.mps.is_available())
 
+sdkKey = os.getenv('sdkKEY')
+
+ldclient.set_config(Config(sdkKey))
+client = ldclient.get()
 
 context = Context.builder("huggie-face") \
     .set("application", "deepSeekChatbot") \

requirements.txt

@@ -6,4 +6,4 @@ minijinja
 torch --extra-index-url https://download.pytorch.org/whl/cu118
 torchvision
 torchaudio
-launchdarkly-server-sdk
+launchdarkly-server-sdk

start.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+
+echo "Setting up environment..."
+
+# Create virtual environment if it doesn't exist
+if [ ! -d "venv" ]; then
+    echo "Creating virtual environment..."
+    python3 -m venv venv
+fi
+
+# Activate virtual environment
+source venv/bin/activate
+
+# Install requirements
+echo "Installing dependencies..."
+pip install -r requirements.txt
+
+# Start the app
+echo "Launching your Hugging Face app!"
+sdkKEY=${sdkKEY} python app.py