Upload 6 files

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+from fastai.vision.all import *
+from huggingface_hub import from_pretrained_fastai
+import torch, os
+
+os.environ.setdefault("OMP_NUM_THREADS", "1")
+torch.set_num_threads(1)
+
+learn = from_pretrained_fastai("joanrc/ST_Classifier")
+try:
+    learn.to_fp32()
+except:
+    pass
+labels = learn.dls.vocab
+
+def predict(img):
+    img = PILImage.create(img)          # same flow as before
+    pred, pred_idx, probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+title = "Stranger Things Dimension Classifier"
+description = "Tries to predict at what dimension the image was taken."
+examples = ['real_world_im', 'the_abbys_im', 'upside_down_im']
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=3),
+    title=title,
+    description=description,
+    examples=examples,
+    cache_examples=False,       # <-- don’t pre-run at startup
+)
+
+demo.queue(max_size=8).launch(show_error=True, debug=True)

packages.txt

@@ -0,0 +1,9 @@
+ffmpeg
+libsm6
+libxext6
+libgl1
+libglx-mesa0
+git
+git-lfs
+cmake
+rsync

requirements.txt

@@ -0,0 +1,12 @@
+fastai==2.7.17
+huggingface_hub[fastai]==0.24.6
+timm==1.0.9
+fasttransform==0.0.2
+cloudpickle==3.0.0
+numpy==1.26.4
+pillow>=10.3.0
+
+torch==2.2.2
+torchvision==0.17.2
+torchaudio==2.2.2
+--extra-index-url https://download.pytorch.org/whl/cpu