soiz1 commited on
Commit
db59c04
·
verified ·
1 Parent(s): 1b2575b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import cv2
 
2
  import os
3
  from PIL import Image
4
  import numpy as np
@@ -6,11 +7,11 @@ import torch
6
  from torch.autograd import Variable
7
  from torchvision import transforms
8
  import torch.nn.functional as F
9
- from flask import Flask, request, jsonify, send_file
10
- import io
11
- from werkzeug.utils import secure_filename
12
  import warnings
13
  warnings.filterwarnings("ignore")
 
14
 
15
  os.system("git clone https://github.com/xuebinqin/DIS")
16
  os.system("mv DIS/IS-Net/* .")
 
1
  import cv2
2
+ import gradio as gr
3
  import os
4
  from PIL import Image
5
  import numpy as np
 
7
  from torch.autograd import Variable
8
  from torchvision import transforms
9
  import torch.nn.functional as F
10
+ import gdown
11
+ import matplotlib.pyplot as plt
 
12
  import warnings
13
  warnings.filterwarnings("ignore")
14
+ import flask
15
 
16
  os.system("git clone https://github.com/xuebinqin/DIS")
17
  os.system("mv DIS/IS-Net/* .")