stay-toxicx commited on
Commit
83f12db
1 Parent(s): 4e17160

update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py CHANGED
@@ -5,6 +5,24 @@ import random
5
  from diffusers import DiffusionPipeline
6
  import torch
7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
  model_repo_id = "stabilityai/sdxl-turbo" #Replace to the model you would like to use
10
 
 
5
  from diffusers import DiffusionPipeline
6
  import torch
7
 
8
+ from gradio_client import Client
9
+ a22= input("Enter prompt!!")
10
+
11
+ client = Client("mukaist/DALLE-4K")
12
+ result = client.predict(
13
+ prompt=a22,
14
+ negative_prompt="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
15
+ use_negative_prompt=True,
16
+ style="3840 x 2160",
17
+ seed=0,
18
+ width=1024,
19
+ height=1024,
20
+ guidance_scale=6,
21
+ randomize_seed=True,
22
+ api_name="/run"
23
+ )
24
+ print(result)
25
+
26
  device = "cuda" if torch.cuda.is_available() else "cpu"
27
  model_repo_id = "stabilityai/sdxl-turbo" #Replace to the model you would like to use
28