Commit 707b1b18 authored by Saswat's avatar Saswat

Fix gui.py

parent fdb0f7ca
import torch import torch
class CFG: class CFG:
debug = True debug = False
seed = 42 seed = 42
# Paths # Paths
......
...@@ -3,11 +3,11 @@ import matplotlib.pyplot as plt ...@@ -3,11 +3,11 @@ import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from tkinter.ttk import * from tkinter.ttk import *
from infer import find_matches from infer import find_matches
from data import make_train_valid_dfs from data import gen_train_valid_dfs
from infer import * from infer import *
from config import CFG 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) model = get_image_embeddings(valid_df, CFG.model_path)
plot = None 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