Upload app/main.py with huggingface_hub

app/main.py

@@ -11,9 +11,12 @@ import urllib.request
 from contextlib import asynccontextmanager
 
 import torch
-# Restrict PyTorch thread-pool memory on single/low-core cloud instances
-torch.set_num_threads(1)
-torch.set_num_interop_threads(1)
+# Restrict PyTorch thread-pool memory if possible
+try:
+    torch.set_num_threads(1)
+    torch.set_num_interop_threads(1)
+except (RuntimeError, ValueError):
+    pass
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware