divyamakkar0
commited on
Commit
·
1a8217a
1
Parent(s):
c8b5862
update
Browse files- config.json +5 -0
- model.pth +1 -1
- model_card.md +14 -0
- requirements.txt +5 -0
config.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": ["CustomModel"],
|
| 3 |
+
"task": "feature-extraction"
|
| 4 |
+
}
|
| 5 |
+
|
model.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1548
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:647bddce52a81b708a96e222e548e2ab2bc44cf5a486537dbaf6be492f06706f
|
| 3 |
size 1548
|
model_card.md
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
tags:
|
| 2 |
+
|
| 3 |
+
- pytorch
|
| 4 |
+
- deep-learning
|
| 5 |
+
- custom-model
|
| 6 |
+
- feature-extraction # Replace with the actual task
|
| 7 |
+
license: apache-2.0
|
| 8 |
+
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# 🚀 Model Name: model-test
|
| 12 |
+
|
| 13 |
+
This is a custom PyTorch model trained for **[task name]** (e.g., regression, classification, etc.).
|
| 14 |
+
It was trained on **[dataset name]** and optimized using **[optimizer]**.
|
requirements.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Flask==3.0.3
|
| 2 |
+
Flask-Cors==4.0.1
|
| 3 |
+
numpy==2.0.0
|
| 4 |
+
torch==2.3.1
|
| 5 |
+
gunicorn==22.0.0
|