Update app.py

app.py

@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 # Copyright 2023 Dmitry Ustalov
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -27,6 +29,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import networkx as nx
 import pandas as pd
+from matplotlib.pyplot import Figure  # type: ignore[attr-defined]
 
 if 'MCL_BIN' in os.environ and os.path.isfile(os.environ['MCL_BIN']) and os.access(os.environ['MCL_BIN'], os.X_OK):
     MCL: Optional[str] = os.environ['MCL_BIN']
@@ -107,7 +110,7 @@ SENSE = re.compile(r'^(?P<item>\d+)#(?P<sense>\d+)$')
 
 
 # noinspection PyPep8Naming
-def visualize(G: nx.Graph, seed: int = 0) -> plt.Figure:
+def visualize(G: nx.Graph, seed: int = 0) -> Figure:
     pos = nx.spring_layout(G, seed=seed)
 
     fig = plt.figure(dpi=240)
@@ -161,7 +164,7 @@ def watset(G: nx.Graph, algorithm: str, seed: int = 0,
         return df_clusters, None
 
 
-def handler(file: IO[bytes], algorithm: str, seed: int) -> Tuple[pd.DataFrame, plt.Figure]:
+def handler(file: IO[bytes], algorithm: str, seed: int) -> Tuple[pd.DataFrame, Figure]:
     if file is None:
         raise gr.Error('File must be uploaded')