Spaces:
Sleeping
Sleeping
update requirements
Browse files- app.py +1 -1
- requirements.txt +2 -0
app.py
CHANGED
|
@@ -32,7 +32,7 @@ class TrainStudent(nn.Module):
|
|
| 32 |
return mse
|
| 33 |
|
| 34 |
chunk_data = generate_chunk_data(["AI","moon","brain"])
|
| 35 |
-
student_model=torch.load("myTextEmbeddingStudent.pt",map_location='cpu').student_model
|
| 36 |
# create the embedding vector database
|
| 37 |
chunk_emb = generate_chunk_emb(student_model, chunk_data)
|
| 38 |
|
|
|
|
| 32 |
return mse
|
| 33 |
|
| 34 |
chunk_data = generate_chunk_data(["AI","moon","brain"])
|
| 35 |
+
student_model=torch.load("myTextEmbeddingStudent.pt",map_location='cpu').student_model.eval()
|
| 36 |
# create the embedding vector database
|
| 37 |
chunk_emb = generate_chunk_emb(student_model, chunk_data)
|
| 38 |
|
requirements.txt
CHANGED
|
@@ -1,3 +1,5 @@
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
pandas
|
|
|
|
|
|
|
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
pandas
|
| 4 |
+
requests
|
| 5 |
+
gradio
|