Hugging Face's logo

Spaces:
arcrao
/
caricature
Runtime error

App Files Community
arcrao commited on
Commit
a869da0
•
1 Parent(s): 61b8c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoModel
6
  app = FastAPI()
7
 
8
  # Load the model once when the application starts
9
- pipe = DiffusionPipeline.from_pretrained("Norod78/caricature-portraits-blip-captions-512", force_download=True)
10
 
11
  @app.post("/generate/")
12
  async def generate_image(prompt: str):
 
6
  app = FastAPI()
7
 
8
  # Load the model once when the application starts
9
+ pipe = DiffusionPipeline.from_pretrained("Norod78/sd15-caricature-portraits-blip-captions")
10
 
11
  @app.post("/generate/")
12
  async def generate_image(prompt: str):