3laa2 commited on
Commit
56e53c8
·
1 Parent(s): e429e65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -16
app.py CHANGED
@@ -10,22 +10,7 @@ def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cuda'):
10
  pipe = StableDiffusionPipeline.from_pretrained(loc, torch_dtype=torch.float16)
11
  pipe = pipe.to(mch)
12
  return pipe
13
-
14
-
15
-
16
- openai.api_key = "please-paste-your-API-key-here"
17
-
18
- def chatWithGPT(prompt):
19
- completion = openai.ChatCompletion.create(
20
- model="gpt-3.5-turbo",
21
- messages=[
22
- {"role": "expert in creating prompts for stable diffusion", "content": prompt}
23
- ]
24
- )
25
- return print(completion.choices[0].message.content)
26
-
27
-
28
-
29
  t2i = st.checkbox("Text2Image")
30
 
31
  if t2i:
 
10
  pipe = StableDiffusionPipeline.from_pretrained(loc, torch_dtype=torch.float16)
11
  pipe = pipe.to(mch)
12
  return pipe
13
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  t2i = st.checkbox("Text2Image")
15
 
16
  if t2i: