Saurav Chaudhari commited on
Commit
025ae61
·
1 Parent(s): cc3131f

Add application file

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import pyiqa
3
  import torch
4
  import pandas as pd
 
5
  from PIL import Image
6
 
7
  # Initialize device
@@ -18,7 +19,8 @@ metrics = {
18
  def analyze_image(input_img):
19
  if input_img is None:
20
  return None
21
-
 
22
  # Standardize input for pyiqa
23
  img_tensor = pyiqa.utils.img2tensor(input_img).unsqueeze(0).to(device)
24
 
 
2
  import pyiqa
3
  import torch
4
  import pandas as pd
5
+ import numpy
6
  from PIL import Image
7
 
8
  # Initialize device
 
19
  def analyze_image(input_img):
20
  if input_img is None:
21
  return None
22
+
23
+ np_img = np.array(input_img)
24
  # Standardize input for pyiqa
25
  img_tensor = pyiqa.utils.img2tensor(input_img).unsqueeze(0).to(device)
26