Update app.py

app.py

@@ -1,5 +1,7 @@
-!pip install transformers
-from transformers import AutoProcessor, AutoModelForImageTextToText
+import gradio as gr
 
-processor = AutoProcessor.from_pretrained("gauravbahl/unsloth_finetune")
-model = AutoModelForImageTextToText.from_pretrained("gauravbahl/unsloth_finetune")
+from transformers import pipeline
+
+pipe = pipeline(model="gauravbahl/unsloth_finetune")
+
+gr.Interface.from_pipeline(pipe).launch()