Updated requirements.txt

Signed-off-by: Kieran Fraser <[email protected]>

app.py

@@ -386,15 +386,16 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
         gr.Markdown("Select a Hugging Face model to launch an adversarial attack against.")
         model_type = gr.Radio(label="Hugging Face Model", choices=["Example", "Other"], value="Example")
         with gr.Column(visible=False) as other_model:
+            gr.Markdown("Coming soon.")
             model_url = gr.Text(label="Model URL",
                     placeholder="e.g. facebook/deit-tiny-distilled-patch16-224",
-                    value='facebook/deit-tiny-distilled-patch16-224')
-            model_input_channels = gr.Text(label="Input channels", value=3)
-            model_input_height = gr.Text(label="Input height", value=32)
-            model_input_width = gr.Text(label="Input width", value=32)
-            model_num_classes = gr.Text(label="Number of classes", value=10)
-            model_clip_values = gr.Radio(label="Clip values", choices=[1, 255], value=1)
-            model_upsample_scaling = gr.Slider(label="Upsample scale factor", minimum=1, maximum=10, value=7)
+                    value='facebook/deit-tiny-distilled-patch16-224', visible=False)
+            model_input_channels = gr.Text(label="Input channels", value=3, visible=False)
+            model_input_height = gr.Text(label="Input height", value=32, visible=False)
+            model_input_width = gr.Text(label="Input width", value=32, visible=False)
+            model_num_classes = gr.Text(label="Number of classes", value=10, visible=False)
+            model_clip_values = gr.Radio(label="Clip values", choices=[1, 255], value=1, visible=False)
+            model_upsample_scaling = gr.Slider(label="Upsample scale factor", minimum=1, maximum=10, value=7, visible=False)
         
         model_type.change(show_params, model_type, other_model)
         

requirements.txt

@@ -1,8 +1,8 @@
 pandas
 jupyter
-torch
-torchvision
 transformers
+torch==1.13.1
+torchvision==0.14.0
 tensorflow==2.10.1; sys_platform != "darwin"
 tensorflow-macos; sys_platform == "darwin"
 tensorflow-metal; sys_platform == "darwin"