emar commited on
Commit
4b52a00
1 Parent(s): 25a6760
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  class Config(BaseModel):
16
  model_config = {'protected_namespaces': ()}
17
 
18
- @spaces.GPU(duration=240)
19
  def setup():
20
  Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5", device=DEVICE)
21
  Settings.llm = HuggingFaceLLM(
@@ -30,7 +30,7 @@ def setup():
30
  setup()
31
 
32
  # Load the existing index
33
- @spaces.GPU
34
  def load_context():
35
  storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
36
  index = load_index_from_storage(storage_context)
 
15
  class Config(BaseModel):
16
  model_config = {'protected_namespaces': ()}
17
 
18
+ # @spaces.GPU(duration=240)
19
  def setup():
20
  Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5", device=DEVICE)
21
  Settings.llm = HuggingFaceLLM(
 
30
  setup()
31
 
32
  # Load the existing index
33
+ # @spaces.GPU
34
  def load_context():
35
  storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
36
  index = load_index_from_storage(storage_context)