Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,8 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
import torch
|
3 |
from t2v_metrics import VQAScore, list_all_vqascore_models
|
4 |
|
5 |
-
torch.jit.script = lambda f: f # Avoid script error in lambda
|
6 |
-
|
7 |
def update_model(model_name):
|
8 |
return VQAScore(model=model_name, device="cuda")
|
9 |
|
|
|
1 |
import gradio as gr
|
2 |
+
torch.jit.script = lambda f: f # Avoid script error in lambda
|
3 |
import torch
|
4 |
from t2v_metrics import VQAScore, list_all_vqascore_models
|
5 |
|
|
|
|
|
6 |
def update_model(model_name):
|
7 |
return VQAScore(model=model_name, device="cuda")
|
8 |
|