divyamakkar0 commited on
Commit
1a8217a
·
1 Parent(s): c8b5862
Files changed (4) hide show
  1. config.json +5 -0
  2. model.pth +1 -1
  3. model_card.md +14 -0
  4. 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:c4956d1399d3b3712d17c57d45b400a2f0bba42669fcc4fb9634eb4d52986e16
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