fix torch import

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import timm
-from torch import nn
+import torch
 
 model = timm.create_model("mobileone_s2", pretrained = False)
-model.head.fc = nn.Linear(model.head.fc.in_features,3)
+model.head.fc = torch.nn.Linear(model.head.fc.in_features,3)
 model.load_state_dict(torch.load("olive_classifier.pth", weights_only=True))
 model.eval()