Update app.py

app.py

@@ -6,6 +6,8 @@ import torch.nn as nn
 from torchvision import models, transforms
 import streamlit as st
 from typing import Tuple
+from fpdf import FPDF
+import io
 
 # Device configuration
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')