Update app.py
Browse files
app.py
CHANGED
|
@@ -34,6 +34,7 @@ from data import get_dataset
|
|
| 34 |
import torchvision.transforms as transforms
|
| 35 |
|
| 36 |
import gradio as gr
|
|
|
|
| 37 |
|
| 38 |
model_name = "convnext_xlarge_in22k"
|
| 39 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
|
| 34 |
import torchvision.transforms as transforms
|
| 35 |
|
| 36 |
import gradio as gr
|
| 37 |
+
import streamlit as st
|
| 38 |
|
| 39 |
model_name = "convnext_xlarge_in22k"
|
| 40 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|