peterpeter8585 commited on
Commit
cc5bc8f
1 Parent(s): 966ab30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import transformers
8
  transformers.utils.move_cache()
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  import os
11
- os.environ["password"]
12
  if torch.cuda.is_available():
13
  torch.cuda.max_memory_allocated(device=device)
14
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
@@ -331,6 +331,6 @@ ab= gr.ChatInterface(
331
  )
332
  if __name__ == "__main__":
333
  with gr.Blocks() as ai:
334
- gr.TabbedInterface([aa, demo2], ["gpt4", "image create"])
335
  ai.queue(max_size=300)
336
  ai.launch()
 
8
  transformers.utils.move_cache()
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  import os
11
+ password1=os.environ["password"]
12
  if torch.cuda.is_available():
13
  torch.cuda.max_memory_allocated(device=device)
14
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
 
331
  )
332
  if __name__ == "__main__":
333
  with gr.Blocks() as ai:
334
+ gr.TabbedInterface([aa, ab, demo2], ["gpt4(Password needed)", "gpt4(only for programming)","image create"])
335
  ai.queue(max_size=300)
336
  ai.launch()