assentian1970 commited on
Commit
7b41a20
·
verified ·
1 Parent(s): ef750f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -7
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
- import os
24
- os.system("nvidia-smi")
 
 
 
 
 
 
 
 
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)