multimodalart HF staff commited on
Commit
d40f6d4
·
1 Parent(s): ec0ec26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -1,6 +1,19 @@
1
  import os
2
  import gradio as gr
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  def update(name):
5
  os.system('''python src/inversion.py \
6
  --input_image "assets/test_images/cats/cat_1.png" \
 
1
  import os
2
  import gradio as gr
3
 
4
+ import os, pdb
5
+
6
+ import argparse
7
+ import numpy as np
8
+ import torch
9
+ import requests
10
+ from PIL import Image
11
+
12
+ from lavis.models import load_model_and_preprocess
13
+
14
+ from src.utils.ddim_inv import DDIMInversion
15
+ from src.utils.scheduler import DDIMInverseScheduler
16
+
17
  def update(name):
18
  os.system('''python src/inversion.py \
19
  --input_image "assets/test_images/cats/cat_1.png" \