base

app.py

@@ -0,0 +1,15 @@
+import numpy as np
+import gradio as gr
+
+def sepia(input_img):
+    sepia_filter = np.array([
+        [0.393, 0.769, 0.189], 
+        [0.349, 0.686, 0.168], 
+        [0.272, 0.534, 0.131]
+    ])
+    sepia_img = input_img.dot(sepia_filter.T)
+    sepia_img /= sepia_img.max()
+    return sepia_img
+
+iface = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
+iface.launch()

requirements.txt

@@ -0,0 +1,9 @@
+# requirements.txt
+torch==2.0.1
+torchvision==0.15.2
+torch-lr-finder==0.2.1
+albumentations==1.3.1
+torchsummary==1.5.1
+grad-cam==1.4.8
+pytorch_lightning==2.0.6
+gradio==3.39.0