update changes

README.md

@@ -10,3 +10,11 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+### Performing inference with the trained model.
+
+#### Run the command below to use gradio to run the model.
+
+```cmd
+python app.py
+```

inference.py

@@ -140,6 +140,7 @@ def deepfakes_image_predict(input_image):
 
     out = nn.Softmax()(multimodal_grads)
     max = torch.argmax(out, dim=-1) #Index of the max value in the tensor.
+    max = max.cpu().detach().numpy()
     max_value = out[max] #Actual value of the tensor.
     max_value = np.argmax(out[max].detach().numpy())