Kaushik Bar commited on
Commit
ff69146
1 Parent(s): 1fd8303

first commit

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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()),