Emil25 commited on
Commit
17301aa
1 Parent(s): 2236871

Update pages/Visualization.py

Browse files
Files changed (1) hide show
  1. pages/Visualization.py +2 -1
pages/Visualization.py CHANGED
@@ -7,7 +7,8 @@ import os
7
  st.set_page_config(page_title="Student Assistant")
8
 
9
  # API для генерации изображения
10
- API_URL_img = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
 
11
  headers = {"Authorization": os.getenv("api_token")}
12
 
13
  # Функция генерации изображения
 
7
  st.set_page_config(page_title="Student Assistant")
8
 
9
  # API для генерации изображения
10
+ #API_URL_img = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
11
+ API_URL_img = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
12
  headers = {"Authorization": os.getenv("api_token")}
13
 
14
  # Функция генерации изображения