| <!DOCTYPE html> |
| <html lang="en"> |
| <head> |
| <meta charset="UTF-8"> |
| <meta name="viewport" content="width=device-width, initial-scale=1.0"> |
| <title>Image Style Transfer</title> |
| </head> |
| <body> |
| <h1>Image Style Transfer</h1> |
| <form id="uploadForm"> |
| <input type="file" id="fileInput" accept="image/*" required> |
| <select id="styleSelect" required> |
| <option value="">Select Style</option> |
| <option value="AnimeGANv3_Arcane">AnimeGANv3_Arcane</option> |
| <option value="AnimeGANv3_Trump v1.0">AnimeGANv3_Trump v1.0</option> |
| <option value="AnimeGANv3_Shinkai">AnimeGANv3_Shinkai</option> |
| |
| </select> |
| <label for="faceCheckbox">Detect Face?</label> |
| <input type="checkbox" id="faceCheckbox"> |
| <button type="submit">Upload & Style Transfer</button> |
| </form> |
|
|
| <div id="resultContainer" style="display: none;"> |
| <h2>Stylized Image</h2> |
| <img id="resultImage" src="" alt="Stylized Image"> |
| </div> |
|
|
| <script> |
| document.getElementById('uploadForm').addEventListener('submit', async function(event) { |
| event.preventDefault(); |
| |
| const fileInput = document.getElementById('fileInput'); |
| const styleSelect = document.getElementById('styleSelect'); |
| const faceCheckbox = document.getElementById('faceCheckbox'); |
| |
| const file = fileInput.files[0]; |
| const style = styleSelect.value; |
| const if_face = faceCheckbox.checked ? "Yes" : "No"; |
| |
| const formData = new FormData(); |
| formData.append('file', file); |
| formData.append('Style', style); |
| formData.append('if_face', if_face); |
| |
| try { |
| const response = await fetch('/inference/', { |
| method: 'POST', |
| body: formData |
| }); |
| |
| if (!response.ok) { |
| throw new Error('Error uploading file'); |
| } |
| |
| const blob = await response.blob(); |
| const imageURL = URL.createObjectURL(blob); |
| |
| document.getElementById('resultImage').src = imageURL; |
| document.getElementById('resultContainer').style.display = 'block'; |
| } catch (error) { |
| console.error('Error:', error); |
| } |
| }); |
| </script> |
| </body> |
| </html> |
|
|
