Dileep7729 commited on
Commit
a64c351
·
verified ·
1 Parent(s): 94d9514

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -74,6 +74,11 @@ iface = gr.Interface(
74
  if __name__ == "__main__":
75
  print("Launching the Gradio interface...")
76
  iface.launch()
 
 
 
 
 
77
 
78
 
79
 
 
74
  if __name__ == "__main__":
75
  print("Launching the Gradio interface...")
76
  iface.launch()
77
+ # Save the fine-tuned model
78
+ model.save_pretrained("fine-tuned-model")
79
+ processor.save_pretrained("fine-tuned-model")
80
+
81
+ print("Model and processor saved locally in the 'fine-tuned-model' directory.")
82
 
83
 
84