Phips commited on
Commit
791c7ad
1 Parent(s): d463c57

Update app.py

Browse files

Change to my self trained models only.
Can still be extended, but then I want the author of that model to ask me, so I know that the other model authors are aware and agree that their model can be used in this space

Files changed (1) hide show
  1. app.py +11 -8
app.py CHANGED
@@ -4,14 +4,17 @@ from gradio_imageslider import ImageSlider
4
  from image_gen_aux import UpscaleWithModel
5
  from image_gen_aux.utils import load_image
6
 
 
 
 
 
 
 
 
 
7
  MODELS = {
8
- "UltraSharp": "Kim2091/UltraSharp",
9
- "DAT X4": "OzzyGT/DAT_X4",
10
- "DAT X3": "OzzyGT/DAT_X3",
11
- "DAT X2": "OzzyGT/DAT_X2",
12
  "RealPLKSR X4": "OzzyGT/4xNomosWebPhoto_RealPLKSR",
13
  "DAT-2 RealWebPhoto X4": "Phips/4xRealWebPhoto_v4_dat2",
14
- "4xRemacri": "OzzyGT/4xRemacri",
15
  }
16
 
17
 
@@ -30,8 +33,8 @@ def clear_result():
30
 
31
 
32
  title = """<h1 align="center">Image Upscaler</h1>
33
- <div align="center">This space is a showcase of the different super resolution models you can use to upscale with the
34
- <a href="https://github.com/asomoza/image_gen_aux">Image Generation Auxiliary Tools</a> library.</div>
35
  """
36
 
37
  with gr.Blocks() as demo:
@@ -42,7 +45,7 @@ with gr.Blocks() as demo:
42
 
43
  model_selection = gr.Dropdown(
44
  choices=list(MODELS.keys()),
45
- value="UltraSharp",
46
  label="Model",
47
  )
48
 
 
4
  from image_gen_aux import UpscaleWithModel
5
  from image_gen_aux.utils import load_image
6
 
7
+ # This uses https://github.com/asomoza/image_gen_aux/blob/main/src/image_gen_aux/upscalers/README.md
8
+ # Also this space has been duplicated from their official huggingface space, https://huggingface.co/spaces/OzzyGT/basic_upscaler
9
+ # They did great work, and I was happy to see them to also use my models :) I thought Id duplicate it and extend it.
10
+ # It basically made me get a pro account so I can make a Zero GPU space. And I will also upload more of my models as a model card now to use here.
11
+
12
+ # Start out with my own models. If others like kim, sirosky, and other model trainers would like their models added here, then thats great.
13
+ # I simply want them to message me first so I know that everythings okay with having their model as a selection here since they are the author of that model. If they want their model on here or not basically.
14
+ # I load models from huggingface model cards though, so the model should be hosted on huggingface.
15
  MODELS = {
 
 
 
 
16
  "RealPLKSR X4": "OzzyGT/4xNomosWebPhoto_RealPLKSR",
17
  "DAT-2 RealWebPhoto X4": "Phips/4xRealWebPhoto_v4_dat2",
 
18
  }
19
 
20
 
 
33
 
34
 
35
  title = """<h1 align="center">Image Upscaler</h1>
36
+ <div align="center">Use this Space to upscale your images, makes use of the
37
+ <a href="https://github.com/asomoza/image_gen_aux">Image Generation Auxiliary Tools</a> library. <br> For now makes use of my self trained models, but can be extended to more models from other authors if they message me.</div>
38
  """
39
 
40
  with gr.Blocks() as demo:
 
45
 
46
  model_selection = gr.Dropdown(
47
  choices=list(MODELS.keys()),
48
+ value="RealPLKSR X4",
49
  label="Model",
50
  )
51