shashwatIDR commited on
Commit
f826074
Β·
verified Β·
1 Parent(s): eeea844

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +2 -12
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
- // Immediately restart once on startup
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
  })();