Spaces:
Running
Running
Update server.js
Browse files
server.js
CHANGED
@@ -13,15 +13,11 @@ import fetch from 'node-fetch';
|
|
13 |
dotenv.config();
|
14 |
|
15 |
// --- Hugging Face Space Auto-Restarter ---
|
16 |
-
const HF_TOKEN = process.env.HF_TOKEN
|
17 |
const SPACE_ID = process.env.HF_SPACE_ID || 'shashwatIDR/vision';
|
18 |
const HF_API_BASE = 'https://huggingface.co/api';
|
19 |
|
20 |
async function restartSpace() {
|
21 |
-
if (!HF_TOKEN) {
|
22 |
-
console.error('[β No HF_TOKEN] Hugging Face token not set. Skipping restart.');
|
23 |
-
return;
|
24 |
-
}
|
25 |
console.log(`[π Attempting restart] ${SPACE_ID}`);
|
26 |
try {
|
27 |
const response = await axios.post(`${HF_API_BASE}/spaces/${SPACE_ID}/restart`, {}, {
|
@@ -54,10 +50,6 @@ async function restartSpace() {
|
|
54 |
}
|
55 |
|
56 |
async function testHFAPI() {
|
57 |
-
if (!HF_TOKEN) {
|
58 |
-
console.error('[β No HF_TOKEN] Hugging Face token not set. Skipping API test.');
|
59 |
-
return false;
|
60 |
-
}
|
61 |
try {
|
62 |
console.log('[π Testing Hugging Face API endpoint...]');
|
63 |
const response = await axios.get(`${HF_API_BASE}/spaces`, {
|
@@ -84,9 +76,7 @@ async function testHFAPI() {
|
|
84 |
if (!apiWorking) {
|
85 |
console.log('[β οΈ Warning] API test failed, but proceeding anyway...');
|
86 |
}
|
87 |
-
//
|
88 |
-
restartSpace();
|
89 |
-
// Then restart every 5 minutes
|
90 |
setInterval(restartSpace, 5 * 60 * 1000);
|
91 |
console.log('[β° Scheduled] Space will restart every 5 minutes');
|
92 |
})();
|
|
|
13 |
dotenv.config();
|
14 |
|
15 |
// --- Hugging Face Space Auto-Restarter ---
|
16 |
+
const HF_TOKEN = process.env.HF_TOKEN
|
17 |
const SPACE_ID = process.env.HF_SPACE_ID || 'shashwatIDR/vision';
|
18 |
const HF_API_BASE = 'https://huggingface.co/api';
|
19 |
|
20 |
async function restartSpace() {
|
|
|
|
|
|
|
|
|
21 |
console.log(`[π Attempting restart] ${SPACE_ID}`);
|
22 |
try {
|
23 |
const response = await axios.post(`${HF_API_BASE}/spaces/${SPACE_ID}/restart`, {}, {
|
|
|
50 |
}
|
51 |
|
52 |
async function testHFAPI() {
|
|
|
|
|
|
|
|
|
53 |
try {
|
54 |
console.log('[π Testing Hugging Face API endpoint...]');
|
55 |
const response = await axios.get(`${HF_API_BASE}/spaces`, {
|
|
|
76 |
if (!apiWorking) {
|
77 |
console.log('[β οΈ Warning] API test failed, but proceeding anyway...');
|
78 |
}
|
79 |
+
// Only schedule periodic restarts (no immediate restart on startup)
|
|
|
|
|
80 |
setInterval(restartSpace, 5 * 60 * 1000);
|
81 |
console.log('[β° Scheduled] Space will restart every 5 minutes');
|
82 |
})();
|