barreloflube commited on
Commit
f48c169
·
1 Parent(s): acfdab2

Refactor import statement in models.py to add gradio module

Browse files
Files changed (1) hide show
  1. tabs/images/models.py +2 -0
tabs/images/models.py CHANGED
@@ -1,5 +1,6 @@
1
  from typing import List, Optional, Dict, Any
2
 
 
3
  from pydantic import BaseModel, field_validator
4
  from PIL import Image
5
 
@@ -42,6 +43,7 @@ class BaseReq(BaseModel):
42
  @field_validator('model', 'negative_prompt', 'embeddings', 'clip_skip', 'controlnet_config')
43
  def check_model(cls, values):
44
  for m in appConfig.IMAGES_MODELS:
 
45
  if m.get('repo_id') == values.get('model'):
46
  loader = m.get('loader')
47
 
 
1
  from typing import List, Optional, Dict, Any
2
 
3
+ import gradio as gr
4
  from pydantic import BaseModel, field_validator
5
  from PIL import Image
6
 
 
43
  @field_validator('model', 'negative_prompt', 'embeddings', 'clip_skip', 'controlnet_config')
44
  def check_model(cls, values):
45
  for m in appConfig.IMAGES_MODELS:
46
+ gr.Info(f'{m.get('repo_id')} {values.get('model')}')
47
  if m.get('repo_id') == values.get('model'):
48
  loader = m.get('loader')
49