Fix all the broken things... so far

Dockerfile

@@ -9,7 +9,7 @@ RUN python -m pip install --upgrade pip && \
     pip install --no-cache-dir --no-warn-script-location --user typing-extensions -U
 
 # Stage 2: Runtime
-FROM nvidia/cuda:11.3.1-cudnn8-runtime
+FROM nvidia/cuda:11.0.3-cudnn8-runtime
 
 RUN rm /etc/apt/sources.list.d/cuda.list
 RUN rm /etc/apt/sources.list.d/nvidia-ml.list
@@ -26,5 +26,8 @@ RUN apt update && \
     apt clean && rm -rf /var/lib/apt/lists/*
 COPY --from=builder /root/.local/lib/python3.7/site-packages /usr/local/lib/python3.7/dist-packages
 COPY app.py app.py
+
+# RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10
+
 CMD ["python3", "app.py"]
 EXPOSE 7860

app.py

@@ -1,14 +1,15 @@
 from fawkes.protection import Fawkes
 import gradio as gr
+import os
 
 def predict(level, img):
   # print(img)
   fwks = Fawkes("extractor_2", '0', 1, mode=level)
   fwks.run_protection([img], format='jpeg')
   splt = img.split(".")
-  # print(os.listdir('/tmp'))
-  return splt[0] + "_cloaked." + splt[1]
+  print(os.listdir('/tmp'))
+  return splt[0] + "_cloaked.jpeg"
 
 gr.Interface(fn=predict, inputs=[gr.components.Dropdown(["low", "mid", "high"], label="Protection Level"),
                                  gr.components.Image(type='filepath')],
-                                 outputs=gr.components.Image(type="pil")).launch(show_error=True)
+                                 outputs=gr.components.Image(type="pil")).launch(show_error=True, server_name="0.0.0.0")