Commit ·
b3fb8ff
1
Parent(s): 1b8fb45
Uploaded requirements.txt
Browse files- app.py +1 -1
- requirements.txt +6 -0
app.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
|
|
| 1 |
from ultralytics import YOLO
|
| 2 |
from PIL import Image
|
| 3 |
import gradio as gr
|
| 4 |
from huggingface_hub import snapshot_download
|
| 5 |
-
import os
|
| 6 |
|
| 7 |
# 1) Change this to your model repo (you already have it)
|
| 8 |
REPO_ID = "Interior-Style-Classification"
|
|
|
|
| 1 |
+
import os
|
| 2 |
from ultralytics import YOLO
|
| 3 |
from PIL import Image
|
| 4 |
import gradio as gr
|
| 5 |
from huggingface_hub import snapshot_download
|
|
|
|
| 6 |
|
| 7 |
# 1) Change this to your model repo (you already have it)
|
| 8 |
REPO_ID = "Interior-Style-Classification"
|
requirements.txt
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ultralytics
|
| 2 |
+
torch
|
| 3 |
+
torchvision
|
| 4 |
+
gradio
|
| 5 |
+
Pillow
|
| 6 |
+
huggingface_hub
|