Hugging Face's logo

Spaces:
Amazingldl
/
Zero-shot_object_detection
Runtime error

App Files Community
Amazingldl commited on
Commit
5195576
•
1 Parent(s): ebbfa8d

Update app.py

Browse files

添加import torch

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import numpy as np
 
3
  from typing import List
4
  from PIL import Image, ImageDraw
5
  from transformers import OwlViTProcessor, OwlViTForObjectDetection
 
1
  import gradio as gr
2
  import numpy as np
3
+ import torch
4
  from typing import List
5
  from PIL import Image, ImageDraw
6
  from transformers import OwlViTProcessor, OwlViTForObjectDetection