Upload 147 files

App_main.py

@@ -1,10 +1,6 @@
-import numpy as np
 import os
-os.system('nvidia-smi')
-os.system('ls /usr/local')
-os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
-os.system('pip install -U openmim')
 os.system('mim install mmcv')
+import numpy as np
 import models
 import gradio as gr
 
@@ -88,12 +84,12 @@ examples = [
 ]
 
 with gr.Blocks() as demo:
-    image_input = gr.inputs.Image(type='pil', label='Input Img')
+    image_input = gr.Image(type='pil', label='Input Img')
     # with gr.Row().style(equal_height=True):
     # image_LR_output = gr.outputs.Image(label='LR Img', type='numpy')
-    image_output = gr.outputs.Image(label='SR Result', type='numpy')
+    image_output = gr.Image(label='SR Result', type='numpy')
     with gr.Row():
-        checkpoint = gr.inputs.Radio(['UC', 'AID'], label='Checkpoint')
+        checkpoint = gr.Radio(['UC', 'AID'], label='Checkpoint')
         scale = gr.Slider(1, 10, value=4.0, step=0.1, label='scale')
 
 io = gr.Interface(fn=sr_func,
@@ -111,6 +107,5 @@ io = gr.Interface(fn=sr_func,
                   allow_flagging='auto',
                   examples=examples,
                   cache_examples=True,
-                  layout="grid"
                   )
 io.launch()

README.md

@@ -4,8 +4,9 @@ emoji: 🐢
 colorFrom: red
 colorTo: yellow
 sdk: gradio
-sdk_version: 3.18.0
+sdk_version: 4.2.0
 app_file: App_main.py
 pinned: false
+python_version: 3.10
 ---
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

models/__init__.py

@@ -3,7 +3,7 @@ from . import edsr, rdn, rcan, swin_backbone
 from . import mlp
 from . import liif
 from . import metasr
-from . import rs_super, siren_modulation, transformer_neck, swin_neck
+from . import rs_super, siren_modulation, transformer_neck
 from . cnn_models import *
 from . import rs_multiscale_super
 from .funsr import FUNSR

models/swin_backbone.py

@@ -1,11 +1,10 @@
 from models import register
 from einops import rearrange
-from mmcv.cnn import build_norm_layer, constant_init, trunc_normal_init
+from mmcv.cnn import build_norm_layer
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint as cp
-from .swin_neck import SwinTransformer, PatchEmbed, SwinBlockSequence
 
 
 @register('Swin_backbone')

requirements.txt

@@ -1,9 +1,7 @@
-tensorboardX
-matplotlib
+torch
+torchvision
+openmim
+gradio
 einops
-pyyaml
-opencv-python
-pandas
-scikit-image
-tqdm
+tensorboardX
 easydict