Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pickle
 import numpy as np
 import torch
+import io
 from torch import nn
 from transformers import AutoModelForSequenceClassification
 from sklearn.pipeline import Pipeline