mlbench123 commited on
Commit
ccd2b10
·
verified ·
1 Parent(s): 101bca1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -1,17 +1,13 @@
1
  import os
2
  import sys
3
- import gradio as gr
4
 
5
- # Ensure current directory is importable
6
  sys.path.append(os.path.dirname(__file__))
7
 
8
- # HF must use transformers (NO Ollama)
9
- os.environ["LOCAL_LLM_PROVIDER"] = "transformers"
10
- os.environ["HF_LLM_MODEL"] = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 
11
 
12
  from gradio_new_rag_app import make_app
13
 
14
  demo = make_app()
15
-
16
- # IMPORTANT: DO NOT call demo.launch()
17
- # HF Spaces will serve `demo` automatically
 
1
  import os
2
  import sys
 
3
 
 
4
  sys.path.append(os.path.dirname(__file__))
5
 
6
+ os.environ.setdefault("LOCAL_LLM_PROVIDER", "transformers")
7
+ os.environ.setdefault("HF_LLM_MODEL", "TinyLlama/TinyLlama-1.1B-Chat-v1.0")
8
+ os.environ.setdefault("DB_XLSX", "database.xlsx")
9
+ os.environ.setdefault("EMB_CACHE", "treatment_embeddings.pkl")
10
 
11
  from gradio_new_rag_app import make_app
12
 
13
  demo = make_app()