Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|