Fixed

.vscode/settings.json

@@ -0,0 +1,17 @@
+{
+    "workbench.colorCustomizations": {
+        "activityBar.activeBackground": "#4f6f14",
+        "activityBar.background": "#4f6f14",
+        "activityBar.foreground": "#e7e7e7",
+        "activityBar.inactiveForeground": "#e7e7e799",
+        "activityBarBadge.background": "#0f3a51",
+        "activityBarBadge.foreground": "#e7e7e7",
+        "sash.hoverBorder": "#4f6f14",
+        "statusBar.background": "#30440c",
+        "statusBar.foreground": "#e7e7e7",
+        "statusBarItem.hoverBackground": "#4f6f14",
+        "statusBarItem.remoteBackground": "#30440c",
+        "statusBarItem.remoteForeground": "#e7e7e7"
+    },
+    "peacock.color": "#30440c"
+}

README.md

@@ -1,13 +1,6 @@
----
-title: Pizza NotPizza
+title: Pizza Not Pizza
 emoji: 🍕
-colorFrom: yellow
-colorTo: red
-sdk: streamlit
-sdk_version: 1.10.0
-app_file: app.py
+colorFrom: red
+colorTo: yellow
+sdk: docker
 pinned: false
-license: openrail
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,16 +1,17 @@
 import streamlit as st
 import torch
 from PIL import Image
+import os
 
 from read import classify
 
 st.title("Pizza & Not Pizza")
 
-device = torch.device("cpu")
-checkpoint = torch.load(r"./best.pth.tar")
-model = checkpoint['model']
-classes = checkpoint['classes']
-tran = checkpoint['transform']
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+checkpoint = torch.load(os.path.join(os.getcwd(), "best.pth.tar"))
+model = checkpoint["model"]
+classes = checkpoint["classes"]
+tran = checkpoint["transform"]
 
 # upload image
 while True:
@@ -19,13 +20,13 @@ while True:
     
     if uploaded_file is not None:
         img = Image.open(uploaded_file)
-        st.image(img, caption='Uploaded Image.', use_column_width=True)
+        st.image(img, caption="Uploaded Image.", use_column_width=True)
         label = classify(model, img, tran, classes, device)
         st.write(label)
         
     elif taking_picture is not None:
         img = Image.open(taking_picture)
-        st.image(img, caption='Uploaded Image.', use_column_width=True)
+        st.image(img, caption="Uploaded Image.", use_column_width=True)
         label = classify(model, img, tran, classes, device)
         st.write(label)
     

requirements.txt

@@ -1,3 +1,4 @@
-Pillow
+pillow
 torch
 torchvision
+streamlit