Update app.py

app.py

@@ -8,6 +8,7 @@ import threading
 import re
 import unittest
 from urllib.parse import urlparse
+import spaces
 
 import pandas as pd
 from selenium import webdriver
@@ -34,6 +35,8 @@ from dotenv import load_dotenv
 from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
+
+@spaces()
 def load_model():  # Define load_model() first
     """
     Loads the openLlama model and tokenizer once and returns the pipeline.
@@ -66,9 +69,10 @@ chat_pipeline = load_model()  # Now call load_model()
 nlp = AutoTokenizer.from_pretrained("bert-base-uncased")
 
 HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+
 if not HUGGINGFACE_TOKEN:
     raise ValueError("HUGGINGFACE_TOKEN is not set in the environment variables.")
-
+add_to_git_credential=True
 login(token=HUGGINGFACE_TOKEN)