Spaces:

DataScienceProject
/

VIT_Demo

Sleeping

benjaminStreltzin commited on Sep 24, 2024

Commit

fd4d84b

verified ·

1 Parent(s): 68c6ddf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from PIL import Image
 from vit_model_test import CustomModel  # Ensure you import the correct class
-from Vit_Training import Custom_VIT_Model
 # Initialize the model
 model = Custom_VIT_Model()

 import gradio as gr
 from PIL import Image
 from vit_model_test import CustomModel  # Ensure you import the correct class
+from vit_Training import Custom_VIT_Model
 # Initialize the model
 model = Custom_VIT_Model()