nktssk commited on
Commit
857662e
·
verified ·
1 Parent(s): 9de6643

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,6 +2,9 @@ import os
2
  import gradio as gr
3
  from transformers import pipeline
4
 
 
 
 
5
  depth_estimator = pipeline(task="depth-estimation",
6
  model="Intel/dpt-hybrid-midas")
7
 
 
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
+ import torch.nn.functional as F
6
+ from PIL import Image
7
+
8
  depth_estimator = pipeline(task="depth-estimation",
9
  model="Intel/dpt-hybrid-midas")
10