MoC-IQA
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -41,7 +41,7 @@ def predict(image):
|
|
| 41 |
|
| 42 |
return "The image quality of the image is: {}".format(round(iq, 4))
|
| 43 |
|
| 44 |
-
|
| 45 |
|
| 46 |
interface = gr.Interface(fn=predict, inputs="image", outputs="text")
|
| 47 |
interface.launch(server_name='127.0.0.1',server_port=8088)
|
|
|
|
| 41 |
|
| 42 |
return "The image quality of the image is: {}".format(round(iq, 4))
|
| 43 |
|
| 44 |
+
os.system("wget -O best_model.pkl https://huggingface.co/Zevin2023/MoC-IQA/resolve/main/Koniq10K_570908.pkl")
|
| 45 |
|
| 46 |
interface = gr.Interface(fn=predict, inputs="image", outputs="text")
|
| 47 |
interface.launch(server_name='127.0.0.1',server_port=8088)
|
requirements.txt
CHANGED
|
@@ -4,6 +4,6 @@ openpyxl==3.0.9
|
|
| 4 |
Pillow==10.0.0
|
| 5 |
scipy==1.11.2
|
| 6 |
timm==0.5.4
|
| 7 |
-
torch==1.11.0
|
| 8 |
tqdm==4.61.2
|
| 9 |
opencv-python
|
|
|
|
| 4 |
Pillow==10.0.0
|
| 5 |
scipy==1.11.2
|
| 6 |
timm==0.5.4
|
| 7 |
+
torch==1.11.0+cu113
|
| 8 |
tqdm==4.61.2
|
| 9 |
opencv-python
|