Update app.py
Browse filesChange 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
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">
|
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="
|
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 |
|