Update index.js

index.js

@@ -207,13 +207,28 @@ async function processScreenshot(req, res, {
 }
 
 const PORT = process.env.PORT || 7860
+function restartSpace() {
+	const { SPACE_ID, HF_TOKEN } = process.env
+	return fetch(`https://huggingface.co/api/spaces/${SPACE_ID}/restart`, {
+		method: 'POST',
+		headers: { Authorization: `Bearer ${HF_TOKEN}` }
+	})
+}
+
 function keepAlive() {
 	var url = `https://${process.env.SPACE_HOST}`
 	if (/(\/\/|\.)undefined\./.test(url)) return
 	setInterval(() => {
-		fetch(url, { headers: { Authorization: "Bearer " + process.env.HF_TOKEN}}).catch(console.error)
+		fetch(url, { headers: { Authorization: "Bearer " + process.env.HF_TOKEN}})
+			.then(response => {
+				if (!response.ok) {
+					restartSpace().catch(console.error)
+				}
+			})
+			.catch(console.error)
 	}, 5 * 1000 * 60)
 }
+
 app.listen(PORT, async () => {
 	console.log('App running on port', PORT)
   console.log(fetch)