Added arguments handling with some command-line options:

app.py

@@ -36,6 +36,7 @@
 import datetime
 import deepfakeecg
 import ecg_plot
+import getopt
 import gradio
 import io
 import matplotlib.pyplot as plt
@@ -245,6 +246,12 @@ def analyze(request: gradio.Request):
    print(data)
 
 
+# ###### Print usage and exit ###############################################
+def usage(exitCode = 0):
+   sys.stdout.write('Usage: ' + sys.argv[0] + ' [-d|--device cpu|cuda] [-v|--version]\n')
+   sys.exit(exitCode)
+
+
 
 # ###### Main program #######################################################
 
@@ -312,6 +319,36 @@ img.logo-image {
 """
 
 
+# ====== Check arguments ====================================================
+try:
+   options, args = getopt.getopt(
+      sys.argv[1:],
+      'd:v',
+      [
+         'device=',
+         'version'
+      ])
+   for option, optarg in options:
+      print(option)
+      if option in ( '-d', '--device' ):
+         runOnDevice = optarg
+      elif option in ( '-v', '--version' ):
+         sys.stdout.write('PyTorch version: ' + torch.__version__ + '\n')
+         sys.stdout.write('CUDA version:    ' + torch.version.cuda + '\n')
+         sys.stdout.write('CUDA available:  ' + ('yes' if torch.cuda.is_available() else 'no') + '\n')
+         sys.stdout.write('Device:          ' + runOnDevice + '\n')
+         sys.exit(1)
+      else:
+         sys.stderr.write('ERROR: Invalid option ' + option + '!\n')
+         sys.exit(1)
+
+except getopt.GetoptError as error:
+   sys.stderr.write('ERROR: ' + str(error) + '\n')
+   usage(1)
+if len(args) > 0:
+   usage(1)
+
+
 # ====== Create GUI =========================================================
 with gradio.Blocks(css = css, theme = gradio.themes.Glass(secondary_hue=gradio.themes.colors.blue)) as gui: