add examples

app.py

@@ -1,18 +1,10 @@
 import gradio as gr
-import torch
 from utils import *
 import cv2
 import numpy as np
 import matplotlib.pyplot as plt
 import io
 
-torch.hub.download_url_to_file(
-    'https://github.com/aalto-ui/aim/raw/aim2/backend/data/tests/input_values/wikipedia.org_website.png',
-    'wikipedia.org_website.png')
-torch.hub.download_url_to_file(
-    'https://github.com/aalto-ui/aim/raw/aim2/backend/data/tests/input_values/aalto.fi_website.png',
-    'aalto.fi_website.png')
-
 
 def inference(img, template, angel):
     color_image = cv2.imread(img.name, cv2.IMREAD_COLOR)
@@ -43,7 +35,7 @@ def inference(img, template, angel):
 title = 'Color Harmonization'
 description = 'Compute Color Harmonization with Different Templates'
 article = "<p style='text-align: center'></p>"
-# examples = [['wikipedia.org_website.png'], ['aalto.fi_website.png']]
+examples = [['./examples/aim_interface.png', "V", 25], ['./examples/esfahan_unsplash.jpeg', "I", 0]]
 # css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 
 gr.Interface(

examples/source.md

@@ -0,0 +1,4 @@
+{
+   "esfahan_unsplash.jpeg":"https://images.unsplash.com/photo-1591139308596-9b663fa6d0a0?ixlib=rb-1.2.1&w=1000&q=80",
+   "aim_interface.png":"https://interfacemetrics.aalto.fi/"
+}

requirements.txt

@@ -1,5 +1,4 @@
 opencv-python-headless ~= 4.2.0.34
-torch
 gradio~=3.0.17
 numpy~=1.21.6
 matplotlib~=3.2.2