Update app.py

app.py

@@ -10,10 +10,17 @@ def _install(pkg):
     subprocess.check_call([sys.executable, "-m", "pip", "install", pkg,
                            "-q", "--root-user-action=ignore"])
 
-for _pkg in ["plotly", "scikit-learn"]:
+# Instala dependências que podem estar faltando no container
+_deps = {
+    "torch":        "torch --index-url https://download.pytorch.org/whl/cpu",
+    "plotly":       "plotly",
+    "sklearn":      "scikit-learn",
+}
+for _mod, _pkg in _deps.items():
     try:
-        __import__(_pkg.replace("-","_").split("[")[0])
+        __import__(_mod)
     except ImportError:
+        print(f"Installing {_pkg}...", flush=True)
         _install(_pkg)
 
 import gradio as gr