Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-#from transformers import AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, TrainingArguments, Trainer, pipeline
 #from llama_cpp import Llama
 from datasets import load_dataset
 import os
 import requests
 
+
 # Replace with the direct image URL
 flower_image_url = "https://i.postimg.cc/hG2FG85D/2.png"