Spaces:
Runtime error
Runtime error
import os | |
import gradio as gr | |
import wandb | |
from my_recommending.app import build_app_blocks, MovieMarkdownGenerator | |
from my_recommending.movielens.lit import MovieLensNonGradientRecommender | |
from my_recommending.movielens.data import MovieLens25m | |
project = "Recommending" | |
tmdb_api_token = os.environ["TMDB_API_TOKEN"] | |
lightning_class = MovieLensNonGradientRecommender | |
config = dict( | |
model_artifact="my_mf_slim_movielens_25m:v9", | |
data_artifact="movielens25m:v4", | |
media_directory="media", | |
n_recommendations_from_imdb_ratings=50, | |
) | |
wandb.init(job_type="app", project=project, config=config) | |
model_artifact = wandb.use_artifact(config["model_artifact"]) | |
checkpoint_path = model_artifact.file() | |
lightning_module = lightning_class.load_from_checkpoint( | |
checkpoint_path, map_location="cpu" | |
) | |
recommender = lightning_module.model.eval() | |
data_artifact = wandb.use_artifact(config["data_artifact"]) | |
data_directory = data_artifact.download() | |
movielens = MovieLens25m(directory=data_directory) | |
movie_markdown_generator = MovieMarkdownGenerator( | |
movielens=movielens, tmdb_api_token=tmdb_api_token | |
) | |
with gr.Blocks() as app: | |
build_app_blocks( | |
recommender=recommender, | |
movie_markdown_generator=movie_markdown_generator, | |
media_directory=config["media_directory"], | |
n_recommendations_from_imdb_ratings=config[ | |
"n_recommendations_from_imdb_ratings" | |
], | |
) | |
app.launch() | |