Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 os.environ["WANDB_DISABLED"] = "true"
 
 import numpy as np
-from PIL import Image
 
 import gradio as gr
 import torch
@@ -116,25 +115,14 @@ config = model.vision_encoder.config
 sf_filename = hf_hub_download("FFZG-cleopatra/M2SA", filename="model.safetensors")
 
 load_model(model, sf_filename) 
-# model.load_state_dict(torch.load(model_args.model_name_or_path+"-finetuned/pytorch_model.bin"))
-
-
-# model = AutoModelForSequenceClassification.from_pretrained(
-#         "FFZG-cleopatra/M2SA",
-#         num_labels=3, id2label=id2label,
-#         label2id=label2id
-#     )
-
-
-
 image_processor = AutoImageProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
 def predict_sentiment(text, image):
     print(text, image)
-    print(dir(image))
+    
     image = read_image(image, mode=ImageReadMode.RGB)
-    # image = transforms.ToTensor()(image).unsqueeze(0)
-    print(image)
+    
+    
     text_inputs = tokenizer(
             text,
             max_length=512,
@@ -152,9 +140,6 @@ def predict_sentiment(text, image):
     pixel_values = image_transformations(image)
     text_inputs["pixel_values"] = pixel_values.unsqueeze(0)
    
-    print(text_inputs)
-    print("its going in ",pixel_values)
-
     prediction = None
     with torch.no_grad():
         outputs = model(**text_inputs)