Victoria31 commited on
Commit
39e3df4
·
verified ·
1 Parent(s): bc0238f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ import os
3
  import re
4
  import requests
5
  import numpy as np
 
 
 
6
 
7
  # Configuration
8
  HF_TOKEN = os.getenv("HF_TOKEN", "").strip() # Make sure your token is set in the Space secrets
 
3
  import re
4
  import requests
5
  import numpy as np
6
+ import torch
7
+ from sklearn.neighbors import NearestNeighbors
8
+ from transformers import AutoTokenizer, AutoModel
9
 
10
  # Configuration
11
  HF_TOKEN = os.getenv("HF_TOKEN", "").strip() # Make sure your token is set in the Space secrets