80cols commited on
Commit
ad565c9
·
verified ·
1 Parent(s): b2ed615

Update predictor.py

Browse files
Files changed (1) hide show
  1. predictor.py +8 -0
predictor.py CHANGED
@@ -4,6 +4,7 @@ import numpy as np
4
  from concrete.ml.deployment import FHEModelClient, FHEModelServer
5
  import logging
6
  import gradio as gr
 
7
 
8
  # Configure logging
9
  logging.basicConfig(level=logging.INFO)
@@ -43,6 +44,13 @@ def predict():
43
  str: The prediction result.
44
  str: A message indicating the status of the prediction.
45
  """
 
 
 
 
 
 
 
46
  global encrypted_data, encrypted_prediction
47
  if encrypted_data is None:
48
  return None, gr.update(value="No encrypted data to predict. Please provide encrypted data ❌")
 
4
  from concrete.ml.deployment import FHEModelClient, FHEModelServer
5
  import logging
6
  import gradio as gr
7
+ import time
8
 
9
  # Configure logging
10
  logging.basicConfig(level=logging.INFO)
 
44
  str: The prediction result.
45
  str: A message indicating the status of the prediction.
46
  """
47
+ start = time.time()
48
+ i = 0
49
+ while time.time() - start < 10:
50
+ i+=1
51
+ if i!=0
52
+ return "TEST", "DOUBLE TEST"
53
+
54
  global encrypted_data, encrypted_prediction
55
  if encrypted_data is None:
56
  return None, gr.update(value="No encrypted data to predict. Please provide encrypted data ❌")