Upload app.py

app.py

@@ -10,6 +10,7 @@ import transformers
 from transformers import BertModel, BertTokenizer
 from PIL import Image
 import requests
+import io
 
 class_names = ['Not Depressed', 'Depressed']
 pt_file = hf_hub_download(repo_id="liangc40/sentimental_analysis", filename="model.pt")
@@ -42,7 +43,9 @@ def analyse(text):
     outputs = model(input_ids = encoding['input_ids'], attention_mask = encoding['attention_mask'])
     _, preds = torch.max(outputs, dim=1)
     face_url = "https://raw.githubusercontent.com/liangc40/ID2223_Sentimental_Analysis_Project/main/Image/"+ str(preds) + ".png"
-    img = Image.open(requests.get(face_url, stream=True).raw)
+    r = requests.get(face_url, stream=True)
+    img = Image.open(io.BytesIO(r.content))
+    #img = Image.open(requests.get(face_url, stream=True).raw)
     #print(preds)
     return img