MrDrmm commited on
Commit
af35444
·
verified ·
1 Parent(s): 755c5c3

Update multit2i.py

Browse files
Files changed (1) hide show
  1. multit2i.py +1 -11
multit2i.py CHANGED
@@ -11,17 +11,7 @@ server_timeout = 2600
11
  inference_timeout = 1400
12
 
13
 
14
- def set_hf_token(token):
15
- os.environ["HF_TOKEN"] = token
16
- return f"Токен установлен: {token[:5]}...{token[-5:]}"
17
-
18
- def generate_image(prompt, width, height):
19
- if "HF_TOKEN" not in os.environ or not os.environ["HF_TOKEN"]:
20
- return "Ошибка: Токен Hugging Face не установлен!"
21
-
22
- model = gr.load("models/digiplay/rRealism_v1.0_riiwa", hf_token=os.environ["HF_TOKEN"])
23
- result = model.predict(prompt=prompt, resolution=f"{width}x{height}")
24
- return result
25
 
26
  lock = RLock()
27
  loaded_models = {}
 
11
  inference_timeout = 1400
12
 
13
 
14
+
 
 
 
 
 
 
 
 
 
 
15
 
16
  lock = RLock()
17
  loaded_models = {}