Update app.py

app.py

@@ -141,8 +141,8 @@ def process_video(video_path, true_label=None):
         result = f"CNN Model Accuracy: {cnn_accuracy:.2f}%\n"
         result += f"QCNN Model Accuracy: {qcnn_accuracy:.2f}%"
     else:
-        result = f"CNN Model Predictions:\nClass 0: {cnn_class0_percent:.2f}%\nClass 1: {cnn_class1_percent:.2f}%\n"
-        result += f"QCNN Model Predictions:\nClass 0: {qcnn_class0_percent:.2f}%\nClass 1: {qcnn_class1_percent:.2f}%"
+        result = f"CNN Model Predictions:\nOriginal: {cnn_class0_percent:.2f}%\nFake: {cnn_class1_percent:.2f}%\n"
+        result += f"QCNN Model Predictions:\nOriginal: {qcnn_class0_percent:.2f}%\nFake: {qcnn_class1_percent:.2f}%"
     return result
 
 def predict(video_input):
@@ -192,7 +192,7 @@ with gr.Blocks() as demo:
             )
         with gr.Column():
             output = gr.Textbox(label="Result")
-            predict_button = gr.Button("Predict", elem_classes="gr-button")
+            predict_button = gr.Button("Predict")
 
     predict_button.click(fn=predict, inputs=video_input, outputs=output)
     demo.launch()