Subindo arquivos3x

app.py

@@ -1,18 +1,22 @@
-import gradio as gr
 import os
 import shutil
-from sklearn.metrics import classification_report, confusion_matrix
-import matplotlib.pyplot as plt
-import seaborn as sns
-import numpy as np
-import io
+import joblib  # Para salvar como .pkl
+import gradio as gr
 import torch
 import torch.nn as nn
 import torch.optim as optim
 from torchvision import datasets, transforms, models
 from torch.utils.data import DataLoader, random_split
 from PIL import Image
-import joblib  # Para salvar como .pkl
+import matplotlib.pyplot as plt
+import seaborn as sns
+import numpy as np
+import io
+from sklearn.metrics import classification_report, confusion_matrix
+
+# Adicionando os comandos para instalar o protobuf-compiler
+os.system('apt-get update')
+os.system('apt-get install -y protobuf-compiler')
 
 # Modelos para seleção
 model_dict = {

requirements.txt

@@ -6,10 +6,11 @@ matplotlib==3.4.3
 seaborn==0.11.2
 Pillow==8.4.0
 joblib==1.1.0
-onnx==1.9.0
+onnx==1.13.0
 onnx-tf==1.8.0
 tensorflow==2.8.0
 tensorflow-addons==0.16.1
 tensorflowjs==3.18.0
 Cython==0.29.24
 protobuf==3.19.4
+