Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,8 @@
|
|
1 |
#!/usr/bin/env python
|
|
|
|
|
|
|
|
|
2 |
import spaces
|
3 |
import torch
|
4 |
@spaces.GPU
|
@@ -6,22 +10,33 @@ def debug():
|
|
6 |
torch.randn(10).cuda()
|
7 |
debug()
|
8 |
|
|
|
|
|
9 |
import gradio as gr
|
10 |
from datetime import datetime
|
11 |
-
import torch
|
12 |
-
from transformers import AutoModel, AutoTokenizer
|
13 |
-
from modelscope.hub.snapshot_download import snapshot_download
|
14 |
from PIL import Image
|
15 |
from decord import VideoReader, cpu
|
|
|
|
|
|
|
|
|
16 |
|
|
|
17 |
import gc
|
18 |
-
import io
|
19 |
import tempfile
|
20 |
from ultralytics import YOLO
|
21 |
import numpy as np
|
22 |
import cv2
|
23 |
-
|
24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
# Load YOLOv11 model (update the path as needed)
|
27 |
YOLO_MODEL = YOLO('best_yolov11.pt')
|
@@ -451,4 +466,4 @@ with gr.Blocks(title="Digital Site Diary") as demo:
|
|
451 |
)
|
452 |
|
453 |
if __name__ == "__main__":
|
454 |
-
demo.launch()
|
|
|
1 |
#!/usr/bin/env python
|
2 |
+
|
3 |
+
#!/usr/bin/env python
|
4 |
+
# encoding: utf-8
|
5 |
+
|
6 |
import spaces
|
7 |
import torch
|
8 |
@spaces.GPU
|
|
|
10 |
torch.randn(10).cuda()
|
11 |
debug()
|
12 |
|
13 |
+
import argparse
|
14 |
+
from transformers import AutoModel, AutoTokenizer
|
15 |
import gradio as gr
|
16 |
from datetime import datetime
|
|
|
|
|
|
|
17 |
from PIL import Image
|
18 |
from decord import VideoReader, cpu
|
19 |
+
import io
|
20 |
+
import os
|
21 |
+
os.system("nvidia-smi")
|
22 |
+
from modelscope.hub.snapshot_download import snapshot_download
|
23 |
|
24 |
+
# Additional imports from snippet 1
|
25 |
import gc
|
|
|
26 |
import tempfile
|
27 |
from ultralytics import YOLO
|
28 |
import numpy as np
|
29 |
import cv2
|
30 |
+
|
31 |
+
# Additional imports from snippet 2
|
32 |
+
import copy
|
33 |
+
import requests
|
34 |
+
import base64
|
35 |
+
import json
|
36 |
+
import traceback
|
37 |
+
import re
|
38 |
+
import modelscope_studio as mgr
|
39 |
+
|
40 |
|
41 |
# Load YOLOv11 model (update the path as needed)
|
42 |
YOLO_MODEL = YOLO('best_yolov11.pt')
|
|
|
466 |
)
|
467 |
|
468 |
if __name__ == "__main__":
|
469 |
+
demo.launch(share=False, debug=True, show_api=False, server_port=args.port, server_name=args.host)
|