Update app.py

app.py

@@ -6,6 +6,9 @@ from transformers import SegformerForImageSegmentation, ResNetForImageClassifica
 from PIL import Image
 from torch.utils.data import Dataset, DataLoader
 import streamlit as st
+import gradio as gr
+
+gr.load("models/nvidia/segformer-b0-finetuned-ade-512-512").launch()
 
 # Function to extract zip files
 def extract_zip(zip_file, extract_to):