Add files

app.py

@@ -0,0 +1,54 @@
+import gradio as gr
+import torch
+import torch.nn as nn
+from torchvision import models, transforms
+from PIL import Image
+import requests
+
+# Load the pre-trained MobileNetV2 model from torchvision
+model = models.mobilenet_v2(pretrained=True)
+device = torch.device("mps")
+model.to(device)
+model.eval()  # Set model to evaluation mode
+
+# Modify the class labels
+url = "https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt"
+response = requests.get(url)
+class_labels = response.text.splitlines()
+class_labels[282] = "FLAG{3883}"  # Modify class name to "FLAG{3883}"
+
+# Preprocessing function to prepare the image
+preprocess = transforms.Compose([
+    transforms.Resize(256),
+    transforms.CenterCrop(224),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
+])
+
+# Function to preprocess the input image
+def preprocess_image(image):
+    image = preprocess(image).unsqueeze(0)  # Add batch dimension
+    return image.to(device)  # Move image to the same device as the model
+
+# Prediction function
+def predict(image):
+    # Load the input file
+    reloaded_img_tensor = torch.load(image).to(torch.device("mps"))
+    
+    # Make predictions
+    output = model(reloaded_img_tensor)
+    predicted_label = class_labels[output.argmax(1, keepdim=True).item()]
+    
+    return predicted_label
+
+# Gradio interface
+iface = gr.Interface(
+    fn=predict,  # Function to call for prediction
+    inputs=gr.File(label="Upload a .pt file"),  # Input: .pt file upload
+    outputs=gr.Textbox(label="Predicted Class"),  # Output: Text showing predicted class
+    title="Vault Challenge 3 - CW",  # Title of the interface
+    description="Upload an image, and the model will predict the class. Try to fool the model into predicting the FLAG using C&W!"
+)
+
+# Launch the Gradio interface
+iface.launch()

attack_image_tensor.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21ef127747638c32fc1895fccd85a4765c0b14a4df7c85cfb69a41b0000ca4be
+size 603352

requirements.txt

@@ -0,0 +1,4 @@
+torch
+torchvision
+Pillow
+requests