Hugging Face's logo

Spaces:
Clifx
/
PS2.ai
Running

App Files Community
Linoy Tsaban commited on
Commit
96afb60
•
1 Parent(s): c633c03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
19
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
20
  sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
21
- blip_processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
22
  blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
23
 
24
 
 
18
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
19
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
20
  sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
21
+ blip_processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
22
  blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
23
 
24