treasuraid commited on
Commit
ed2381d
·
1 Parent(s): 4020774

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,9 +15,9 @@ LoRa_DIR = 'VESSL/lora-fintuning-bayc'
15
  DATASET_REPO = 'VESSL/Bored_Ape_NFT_text'
16
  SAMPLE_IMAGE = 'weights/Sample.png'
17
 
 
18
  def load_pipeline_w_lora() :
19
 
20
-
21
  # Load pipeline
22
  pipeline = DiffusionPipeline.from_pretrained(
23
  MODEL_REPO,
 
15
  DATASET_REPO = 'VESSL/Bored_Ape_NFT_text'
16
  SAMPLE_IMAGE = 'weights/Sample.png'
17
 
18
+ @st.cache()
19
  def load_pipeline_w_lora() :
20
 
 
21
  # Load pipeline
22
  pipeline = DiffusionPipeline.from_pretrained(
23
  MODEL_REPO,