sergeipetrov
/

pix2pix-instruct-IE

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

sergeipetrov HF staff commited on Feb 9

Commit

63043dc

•

1 Parent(s): 9c2101e

Update handler.py

Files changed (1) hide show

handler.py +2 -0

handler.py CHANGED Viewed

@@ -4,6 +4,8 @@ from io import BytesIO
 import torch
 import base64
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
 import logging
 import subprocess
 import sys

 import torch
 import base64
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+import diffusers
+import transformers
 import logging
 import subprocess
 import sys