Upload app.py

app.py

@@ -169,12 +169,9 @@ def decrypt_prediction(user_id):
     predictions = fhe_api.deserialize_decrypt_dequantize(encrypted_prediction)
 
 
-if __name__ == "__main__":
-    import gradio as gr
-
 
-def process_pipeline(task, test_file, user_id):
-    eval_key = keygen(task, user_id)
+def process_pipeline(test_file, user_id):
+    eval_key = keygen(user_id)
     encodings = encode_quantize_encrypt(test_file, eval_key)
     encrypted_quantized_encoding = run_fhe(encodings)
     encrypted_prediction = decrypt_prediction(encrypted_quantized_encoding)
@@ -182,13 +179,11 @@ def process_pipeline(task, test_file, user_id):
     return eval_key, encodings, encrypted_quantized_encoding, encrypted_prediction
 
 if __name__ == "__main__":
-    import gradio as gr
+
     app = gr.Interface(
         fn=process_pipeline,
         inputs=[
-            gr.Textbox(label="Task", value="malware"),
             gr.File(label="Test File"),
-            gr.Textbox(label="User ID"),
         ],
         outputs=[
             gr.Textbox(label="Evaluation Key"),
@@ -200,4 +195,4 @@ if __name__ == "__main__":
         description="This is a FHE Model",
     )
 
-    app.launch()
+    app.launch(share=True)