Upload app.py

app.py

@@ -8,6 +8,8 @@ import torch
 from huggingface_hub import hf_hub_download
 import transformers
 from transformers import BertModel, BertTokenizer
+from PIL import Image
+import requests
 
 class_names = ['Not Depressed', 'Depressed']
 pt_file = hf_hub_download(repo_id="liangc40/sentimental_analysis", filename="model.pt")
@@ -36,11 +38,12 @@ def analyse(text):
                                      pad_to_max_length=True,
                                      return_attention_mask=True,
                                      return_tensors='pt')
-
+    face_url = "https://raw.githubusercontent.com/liangc40/ID2223_Sentimental_Analysis_Project/main/Image/"+ str(preds) + ".png"
     outputs = model(input_ids = encoding['input_ids'], attention_mask = encoding['attention_mask'])
     _, preds = torch.max(outputs, dim=1)
+    img = Image.open(requests.get(face_url, stream=True).raw)
     #print(preds)
-    return preds
+    return img
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Sentiment Analysis with Fine-tuned BERT Model")