Update app.py

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import torch
-from PIL import Image
-import io
-import os
-import torch.nn as nn
 import timm
 from torch import nn
 from torchvision import transforms
+from PIL import Image
 import torchvision.models as models
+
+
 # Create the 'model' directory if it doesn't exist (for saving the model later)
 if not os.path.exists('model'):
     os.makedirs('model')