Commit 707b1b18 authored by Saswat's avatar Saswat

Fix gui.py

parent fdb0f7ca
import torch
class CFG:
debug = True
debug = False
seed = 42
# Paths
......
......@@ -3,11 +3,11 @@ import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from tkinter.ttk import *
from infer import find_matches
from data import make_train_valid_dfs
from data import gen_train_valid_dfs
from infer import *
from config import CFG
_, valid_df = make_train_valid_dfs()
_, valid_df = gen_train_valid_dfs()
model = get_image_embeddings(valid_df, CFG.model_path)
plot = None
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment