Kitasurfer commited on
Commit
4bc0f3a
·
verified ·
1 Parent(s): e7e4f84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -15
app.py CHANGED
@@ -1,18 +1,10 @@
1
- import gradio as gr
2
- from diffusers import StableDiffusionPipeline
3
- import torch
4
 
5
- # Используем модель lllyasviel/flux1-dev-bnb-nf4
6
- model_id = "lllyasviel/flux1-dev-bnb-nf4"
7
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
 
9
- def generate_image(prompt):
10
- image = pipe(prompt).images[0]
11
- return image
12
 
13
- gr.Interface(
14
- fn=generate_image,
15
- inputs="text",
16
- outputs="image",
17
- title="Stable Diffusion WebUI"
18
- ).launch()
 
1
+ import os
 
 
2
 
3
+ # Клонируем репозиторий
4
+ os.system("git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git")
 
5
 
6
+ # Перемещаемся в директорию с клонированным репозиторием
7
+ os.chdir("stable-diffusion-webui-forge")
 
8
 
9
+ # Выполняем другие необходимые команды, например, установка зависимостей
10
+ os.system("pip install -r requirements.txt")