Update app.py
Browse files
app.py
CHANGED
|
@@ -15,6 +15,7 @@ import subprocess
|
|
| 15 |
|
| 16 |
import gradio as gr
|
| 17 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 18 |
from datasets import load_dataset
|
| 19 |
import torch
|
| 20 |
import torchvision.transforms as T
|
|
|
|
| 15 |
|
| 16 |
import gradio as gr
|
| 17 |
from huggingface_hub import hf_hub_download
|
| 18 |
+
from safetensors.torch import load_model
|
| 19 |
from datasets import load_dataset
|
| 20 |
import torch
|
| 21 |
import torchvision.transforms as T
|