Update main.py

main.py

@@ -1,17 +1,16 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from PIL import Image
 import torch
 from torchvision import transforms
 from io import BytesIO
-from torch.nn import Module, Conv2d, Linear, BatchNorm2d
-import torch.nn as nn
-import torch.nn.functional as F
-import os
 import base64
 import io
 
+import torch.nn as nn
+import torch.nn.functional as F
+
 class CNN(nn.Module):
     def __init__(self, in_channels, out_channels):
         super(CNN, self).__init__()
@@ -65,7 +64,6 @@ html_form = """
                         processData: false,
                         success: function(response) {
                             $("#prediction").text("Prediction: " + response.prediction);
-
                             // Visualize the uploaded image
                             var imageUrl = "data:image/jpeg;base64," + response.image_base64;
                             $("#uploaded-image").html('<img src="' + imageUrl + '" alt="Uploaded Image" style="max-width: 400px; max-height: 400px;">');
@@ -97,7 +95,6 @@ html_form = """
 async def home():
     return HTMLResponse(content=html_form, status_code=200)
 
-
 def image_to_base64(image):
     # Convert the Image object to bytes
     img_byte_array = io.BytesIO()