shivansh-ka commited on
Commit
22feff3
·
1 Parent(s): b29791f

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +3 -2
app2.py CHANGED
@@ -10,8 +10,9 @@ global prediction
10
  def model_obj():
11
  model = ModelLoader()
12
  prediction = PredictionServices(model.Model, model.Tokenizer)
13
-
14
- model_obj()
 
15
 
16
  def single_predict(text):
17
  preds = prediction.single_predict(text)
 
10
  def model_obj():
11
  model = ModelLoader()
12
  prediction = PredictionServices(model.Model, model.Tokenizer)
13
+ return prediction
14
+
15
+ prediction = model_obj()
16
 
17
  def single_predict(text):
18
  preds = prediction.single_predict(text)