Update app.py

app.py

@@ -1,6 +1,22 @@
 from fastai.vision.all import *
 import gradio as gr
 import glob
+import timm
+from timm.models import convnext
+convnext_model = 'convnext_tiny_in22k'
+model_architecture=timm.create_model(convnext_model)
+
+import torch
+class FastaiConvNext(torch.nn.Module):
+    def __init__(self, original_model):
+        super().__init__()
+        self.features = original_model
+
+    def forward(self, x):
+        x = self.features(x)
+        return x
+
+model = FastaiConvNext(model_architecture)
 
 #class Hook():
 #  def hook_func(self, m, i, o): self.stored = o.detach().clone()