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 |  | 

