Update app.py

app.py

@@ -1,3 +1,17 @@
+try:
+    import albumentations
+except ImportError:
+    os.system('pip install albumentations')
+    import albumentations
+
+try:
+    import toml
+except ImportError:
+    os.system('pip install toml')
+    import toml
+
+os.system('pip install -U gradio')
+
 import gradio as gr
 import torchvision.transforms as transforms
 import random
@@ -13,20 +27,6 @@ from fastai.data.all import *
 from fastai.callback import *
 from pathlib import Path
 
-try:
-    import albumentations
-except ImportError:
-    os.system('pip install albumentations')
-    import albumentations
-
-try:
-    import toml
-except ImportError:
-    os.system('pip install toml')
-    import toml
-
-
-
 #Primero definimos todas las funciones, clases y variables que sopn necesarias para que esto funcione
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def transform_image(image):