Update app.py
Browse files
app.py
CHANGED
|
@@ -21,6 +21,9 @@ import matplotlib.pyplot as plt
|
|
| 21 |
import numpy as np
|
| 22 |
from PIL import Image
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
safe_tensors = "model.safetensors" #hf_hub_download(repo_id="subhuatharva/swim-224-base-satellite-image-classification", filename="model.safetensors")
|
| 25 |
|
| 26 |
model_name = 'swin_s3_base_224'
|
|
|
|
| 21 |
import numpy as np
|
| 22 |
from PIL import Image
|
| 23 |
|
| 24 |
+
import subprocess
|
| 25 |
+
subprocess.run('pip install datasets timm ,shell=True)
|
| 26 |
+
|
| 27 |
safe_tensors = "model.safetensors" #hf_hub_download(repo_id="subhuatharva/swim-224-base-satellite-image-classification", filename="model.safetensors")
|
| 28 |
|
| 29 |
model_name = 'swin_s3_base_224'
|