Upload folder using huggingface_hub

.github/workflows/update_space.yml

@@ -1,28 +1,27 @@
 name: Run Python script
 
 on:
-  push:
-    branches:
-      - main
+    push:
+        branches:
+            - main
 
 jobs:
-  build:
-    runs-on: ubuntu-latest
+    build:
+        runs-on: ubuntu-latest
 
-    steps:
-    - name: Checkout
-      uses: actions/checkout@v2
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v2
 
-    - name: Set up Python
-      uses: actions/setup-python@v2
-      with:
-        python-version: '3.9'
+            - name: Set up Python
+              uses: actions/setup-python@v2
+              with:
+                  python-version: '3.9'
 
-    - name: Install Gradio
-      run: python -m pip install gradio
+            - name: Install Gradio
+              run: python -m pip install gradio
 
-    - name: Log in to Hugging Face
-      run: python -c 'import huggingface_hub; huggingface_hub.login(token="${{ secrets.hf_token }}")'
-
-    - name: Deploy to Spaces
-      run: gradio deploy
+            - name: Deploy to Spaces
+              env:
+                  HF_TOKEN: ${{ secrets.HF_TOKEN }}
+              run: gradio deploy

app.py

@@ -46,7 +46,6 @@ model = None
 tokenizer = None
 
 
-@spaces.GPU
 def load_steering_model():
     """Load the model and tokenizer for steering."""
     global model, tokenizer
@@ -511,6 +510,7 @@ def classify_message(message):
     return best_category, confidence, suggested_brand, reasoning
 
 
+@spaces.GPU
 def generate_steered_response(
     message, history, steering_coeff, max_tokens, brand, steering_type
 ):
@@ -724,7 +724,7 @@ def chat(
 
 
 # Build the Gradio interface
-with gr.Blocks(title="shillLM", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="shillLM") as demo:
     gr.Markdown(
         """
         # 💸 shiLLM 💸        
@@ -1300,4 +1300,6 @@ if __name__ == "__main__":
     print("   Open http://localhost:7860 in your browser to start chatting!")
     print("=" * 50)
 
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    demo.launch(
+        server_name="0.0.0.0", server_port=7860, share=True, theme=gr.themes.Soft()
+    )