akhaliq HF Staff commited on
Commit
c7bd838
·
verified ·
1 Parent(s): d547046

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +26 -0
  2. requirements.txt +20 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import spaces
2
+ from dataclasses import dataclass
3
+ import json
4
+ import logging
5
+ import os
6
+ import random
7
+ import re
8
+ import sys
9
+ import warnings
10
+
11
+ from PIL import Image
12
+ from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
13
+ import gradio as gr
14
+ import torch
15
+ from transformers import AutoModelForCausalLM, AutoTokenizer
16
+
17
+ from prompt_check import is_unsafe_prompt
18
+
19
+ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
20
+
21
+ from diffusers import ZImagePipeline
22
+ from diffusers.models.transformers.transformer_z_image import ZImageTransformer2DModel
23
+
24
+ from pe import prompt_template
25
+
26
+ #
requirements.txt ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ git+https://github.com/huggingface/diffusers
2
+ git+https://github.com/huggingface/transformers
3
+ gradio
4
+ Pillow
5
+ torch
6
+ spaces
7
+ sentencepiece
8
+ accelerate
9
+ tokenizers
10
+ datasets
11
+ numpy
12
+ scipy
13
+ requests
14
+ torchvision
15
+ torchaudio
16
+ safetensors
17
+ opencv-python
18
+ pandas
19
+ prompt-check
20
+ pe