Update app.py

app.py

@@ -1,5 +1,20 @@
+import gradio as gr
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import re
-from transformers import pipeline
+
+# Load the fine-tuned model from Hugging Face Hub
+model_name = "redietmolla/amharic_qa_fine_tuned_llama_model"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Set up the text generation pipeline
+text_generation_pipeline = pipeline(
+    task="text-generation", 
+    model=model, 
+    tokenizer=tokenizer, 
+    max_length=300
+)
+
 
 # Function to clean and truncate the generated text after encountering unwanted content
 def clean_and_truncate_text(text):