MRK / app.py
YousefKhamis's picture
Update app.py
1e60ba1 verified
Raw
History Blame Contribute Delete
1.04 kB
import numpy as np
# =========================
# ๐ŸŸก ุชุญูˆูŠู„ ู…ู„ู .h ุฅู„ู‰ model.tflite
# =========================
def h_to_tflite_binary():
from mrk_model import mrk_model_data, mrk_model_data_len
with open("model.tflite", "wb") as f:
f.write(bytes(mrk_model_data[:mrk_model_data_len]))
print("โœ” model.tflite created from .h")
# =========================
# ๐Ÿ“ฆ ุชุญู…ูŠู„ ุงู„ู…ูˆุฏูŠู„ (ุจุฏูˆู† TensorFlow)
# =========================
def load_model():
try:
with open("model.tflite", "rb") as f:
data = f.read()
print("โœ” model loaded, size:", len(data))
return data
except:
print("โš  model.tflite not found, creating from .h...")
h_to_tflite_binary()
with open("model.tflite", "rb") as f:
return f.read()
# =========================
# ๐Ÿš€ MAIN
# =========================
if __name__ == "__main__":
model = load_model()
# ุงุฎุชุจุงุฑ ุจุณูŠุท (ู…ุด inference ุญู‚ูŠู‚ูŠ)
print("First bytes:", model[:20])