CloudAnts commited on
Commit
efa5bc0
·
1 Parent(s): 4557200
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -14,7 +14,10 @@ from ultralytics import YOLOv10
14
  import supervision as sv
15
  import supervision as sv
16
  from flask import Flask, request, jsonify, send_from_directory, render_template
17
-
 
 
 
18
  import textwrap
19
  app = Flask(__name__)
20
 
 
14
  import supervision as sv
15
  import supervision as sv
16
  from flask import Flask, request, jsonify, send_from_directory, render_template
17
+ print(torch.__version__)
18
+ print(torchvision.__version__)
19
+ import torch
20
+ print(torch.cuda.is_available())
21
  import textwrap
22
  app = Flask(__name__)
23