evangelosmeklis commited on
Commit
8227b96
·
1 Parent(s): bd61f34

fix imports

Browse files
Files changed (2) hide show
  1. drone/drone_chat.py +4 -4
  2. drone/drone_control.py +1 -1
drone/drone_chat.py CHANGED
@@ -7,11 +7,11 @@ import numpy as np
7
  import matplotlib.pyplot as plt
8
  import io
9
  import base64
10
- from hf_model import HfApiModel
11
  import time
12
  # Import compatibility fix for collections.MutableMapping
13
- import compatibility_fix
14
- import drone_control # Import our new drone_control module
15
 
16
  # Set page config at module level - must be first Streamlit command
17
  st.set_page_config(
@@ -509,7 +509,7 @@ def create_qwen_model():
509
  # Return a placeholder model that returns a fixed response
510
  class PlaceholderModel:
511
  def __call__(self, *args, **kwargs):
512
- from hf_model import Message
513
  return Message("Authentication error: No Hugging Face API token provided. Please set an API token to use this feature.")
514
  return PlaceholderModel()
515
 
 
7
  import matplotlib.pyplot as plt
8
  import io
9
  import base64
10
+ from .hf_model import HfApiModel
11
  import time
12
  # Import compatibility fix for collections.MutableMapping
13
+ from . import compatibility_fix
14
+ from . import drone_control # Import our new drone_control module
15
 
16
  # Set page config at module level - must be first Streamlit command
17
  st.set_page_config(
 
509
  # Return a placeholder model that returns a fixed response
510
  class PlaceholderModel:
511
  def __call__(self, *args, **kwargs):
512
+ from .hf_model import Message
513
  return Message("Authentication error: No Hugging Face API token provided. Please set an API token to use this feature.")
514
  return PlaceholderModel()
515
 
drone/drone_control.py CHANGED
@@ -6,7 +6,7 @@ This module provides functions for controlling real drones using DroneKit-Python
6
  import time
7
  import math
8
  # Import compatibility fix for collections.MutableMapping
9
- import compatibility_fix
10
  from dronekit import connect, VehicleMode, LocationGlobalRelative, Command
11
  from pymavlink import mavutil
12
  from typing import Dict, List, Optional, Tuple, Union
 
6
  import time
7
  import math
8
  # Import compatibility fix for collections.MutableMapping
9
+ from . import compatibility_fix
10
  from dronekit import connect, VehicleMode, LocationGlobalRelative, Command
11
  from pymavlink import mavutil
12
  from typing import Dict, List, Optional, Tuple, Union