Kaushik Bar
commited on
Commit
·
ff69146
1
Parent(s):
1fd8303
first commit
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import datetime
|
2 |
import gradio as gr
|
|
|
3 |
import fasttext, torch, clip
|
4 |
from sentence_transformers import SentenceTransformer, util
|
5 |
|
@@ -68,6 +69,7 @@ def prep_examples():
|
|
68 |
return examples
|
69 |
|
70 |
def detect_lang(text):
|
|
|
71 |
seq_lang = 'en'
|
72 |
|
73 |
text = text.replace('\n', ' ')
|
@@ -122,6 +124,8 @@ def sequence_to_classify(text, labels):
|
|
122 |
|
123 |
predicted_labels = list(preds['labels'])
|
124 |
predicted_scores = list(preds['scores'])
|
|
|
|
|
125 |
output = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
|
126 |
print("Date:{}, Sequence:{}, Labels: {}".format(
|
127 |
str(datetime.datetime.now()),
|
|
|
1 |
import datetime
|
2 |
import gradio as gr
|
3 |
+
from langdetect import DetectorFactory
|
4 |
import fasttext, torch, clip
|
5 |
from sentence_transformers import SentenceTransformer, util
|
6 |
|
|
|
69 |
return examples
|
70 |
|
71 |
def detect_lang(text):
|
72 |
+
DetectorFactory.seed = 0
|
73 |
seq_lang = 'en'
|
74 |
|
75 |
text = text.replace('\n', ' ')
|
|
|
124 |
|
125 |
predicted_labels = list(preds['labels'])
|
126 |
predicted_scores = list(preds['scores'])
|
127 |
+
print(predicted_labels)
|
128 |
+
print(predicted_scores)
|
129 |
output = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
|
130 |
print("Date:{}, Sequence:{}, Labels: {}".format(
|
131 |
str(datetime.datetime.now()),
|