Mohammadreza Ghaffarzadeh
commited on
Commit
·
328df23
1
Parent(s):
ca1d264
add requirements
Browse files- app.py +1 -1
- requirements.txt +5 -0
app.py
CHANGED
@@ -2,7 +2,7 @@ import numpy as np
|
|
2 |
import gradio as gr
|
3 |
from transformers import AutoImageProcessor, AutoModelForImageClassification
|
4 |
from PIL import Image
|
5 |
-
|
6 |
|
7 |
|
8 |
processor = AutoImageProcessor.from_pretrained("microsoft/swin-tiny-patch4-window7-224")
|
|
|
2 |
import gradio as gr
|
3 |
from transformers import AutoImageProcessor, AutoModelForImageClassification
|
4 |
from PIL import Image
|
5 |
+
|
6 |
|
7 |
|
8 |
processor = AutoImageProcessor.from_pretrained("microsoft/swin-tiny-patch4-window7-224")
|
requirements.txt
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
torch
|
2 |
+
transformers
|
3 |
+
gradio
|
4 |
+
PIL
|
5 |
+
numpy
|