AlexWortega commited on
Commit
3ee535c
1 Parent(s): f9bc344

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ from rudalle.utils import seed_everything
5
  import sys
6
  from rudolph.model.utils import get_i2t_attention_mask, get_t2t_attention_mask
7
  from rudolph.model import get_rudolph_model, ruDolphModel, FP16Module
8
- from rudolph.pipelines import generate_codebooks, self_reranking_by_image, self_reranking_by_text, show, generate_captions, generate_texts from rudolph.pipelines import zs_clf
 
9
  import gradio as gr
10
  from rudolph import utils
11
  device = 'cuda'
 
5
  import sys
6
  from rudolph.model.utils import get_i2t_attention_mask, get_t2t_attention_mask
7
  from rudolph.model import get_rudolph_model, ruDolphModel, FP16Module
8
+ from rudolph.pipelines import generate_codebooks, self_reranking_by_image, self_reranking_by_text, show, generate_captions, generate_texts
9
+ from rudolph.pipelines import zs_clf
10
  import gradio as gr
11
  from rudolph import utils
12
  device = 'cuda'