Supeem commited on
Commit
3aeec6c
·
1 Parent(s): 488113b
Files changed (1) hide show
  1. main.py +1 -4
main.py CHANGED
@@ -1,13 +1,10 @@
1
- print("NA")
2
  from flask import Flask, request
3
  from flask_cors import CORS, cross_origin
4
  import torch
5
  import model
6
  import numpy as np
7
- print("HEE")
8
  from sentence_transformers import SentenceTransformer
9
- print("KUY")
10
- print("WOW")
11
 
12
  embedding_dim = 384
13
  hidden_dim = 512
 
 
1
  from flask import Flask, request
2
  from flask_cors import CORS, cross_origin
3
  import torch
4
  import model
5
  import numpy as np
 
6
  from sentence_transformers import SentenceTransformer
7
+ sentence_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
8
 
9
  embedding_dim = 384
10
  hidden_dim = 512