mlbench123 commited on
Commit
e4d7abb
·
verified ·
1 Parent(s): aaa22f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -10
app.py CHANGED
@@ -1,19 +1,17 @@
1
  import os
2
  import sys
 
3
 
4
- # Ensure this folder is importable on HF (Linux)
5
  sys.path.append(os.path.dirname(__file__))
6
 
7
- # HF Spaces: do NOT use Ollama; use transformers backend
8
- os.environ.setdefault("LOCAL_LLM_PROVIDER", "transformers")
9
-
10
- # CPU-friendly default model for Spaces
11
- os.environ.setdefault("HF_LLM_MODEL", "TinyLlama/TinyLlama-1.1B-Chat-v1.0")
12
-
13
- # Data file defaults (you will upload these)
14
- os.environ.setdefault("DB_XLSX", "database.xlsx")
15
- os.environ.setdefault("EMB_CACHE", "treatment_embeddings.pkl")
16
 
17
  from gradio_new_rag_app import make_app
18
 
19
  demo = make_app()
 
 
 
 
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