bug fix

app.py

@@ -3,11 +3,7 @@ import pandas as pd
 from matplotlib import pyplot as plt
 
 from config import CONFIG
-from machine_learning.transformers.MakingGraphsAccessible.data import (
-    get_extra_tokens,
-    BenetechOutput,
-    ChartType,
-)
+from data import get_extra_tokens, BenetechOutput, ChartType
 from model import predict_string, build_model
 
 
@@ -46,7 +42,7 @@ def main():
     interface = gradio.Interface(
         title="Making graphs accessible",
         description="Generate textual representation of a graph\n"
-                    "https://www.kaggle.com/competitions/benetech-making-graphs-accessible",
+        "https://www.kaggle.com/competitions/benetech-making-graphs-accessible",
         fn=lambda image: predict_string(image, model),
         inputs="image",
         outputs="text",

data.py

@@ -229,7 +229,7 @@ class AnnotatedImage:
 
 def generate_annotated_images():
     for image_id in tqdm.autonotebook.tqdm(
-            load_train_image_ids(), "Iterating over annotated images"
+        load_train_image_ids(), "Iterating over annotated images"
     ):
         yield AnnotatedImage.from_image_id(image_id)
 
@@ -331,7 +331,7 @@ def convert_number_to_scientific_string(value: int or float) -> str:
 
 
 def convert_axis_data_to_string(
-        axis_data: list[str or float], values_type: ValuesType
+    axis_data: list[str or float], values_type: ValuesType
 ) -> str:
     formatted_axis_data = []
     for value in axis_data: