DevNumb commited on
Commit
7d2d686
·
verified ·
1 Parent(s): 42637b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -1,15 +1,14 @@
1
  import os
2
  import gradio as gr
3
- import fitz # PyMuPDF for PDFs
4
  import docx
5
- import request
6
- HF_TOKEN = os.getenv("HF_TOKEN")
7
- # Load embedding model (fast & free)
8
- API_URL = "https://router.huggingface.co/hf-inference/models/sentence-transformers/all-MiniLM-L6-v2/pipeline/sentence-similarity"
9
- headers = {
10
- "Authorization": f"Bearer {os.environ['HF_TOKEN']}",
11
- }
12
 
 
 
 
 
13
 
14
 
15
  # ---- Text extraction ----
@@ -78,3 +77,4 @@ demo = gr.Interface(
78
  if __name__ == "__main__":
79
  demo.launch()
80
 
 
 
1
  import os
2
  import gradio as gr
3
+ import fitz # PyMuPDF
4
  import docx
5
+ import requests
6
+ import numpy as np
 
 
 
 
 
7
 
8
+ HF_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
9
+ HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
10
+ HF_TOKEN = os.getenv("HF_TOKEN") # set this secret in your Space
11
+ HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
12
 
13
 
14
  # ---- Text extraction ----
 
77
  if __name__ == "__main__":
78
  demo.launch()
79
 
80
+