Spaces:
Runtime error
Runtime error
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 |
|