Upload 2 files

app.py

@@ -1,6 +1,8 @@
 
 import requests
 import time
+from PIL import Image
+import gradio as gr
 
 
 hold_time = time.time()
@@ -11,6 +13,8 @@ headers = {
 	"Content-Type": "application/json" 
 }
 
+
+
 def query(payload):
 	global hold_time
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -22,46 +26,112 @@ def query(payload):
 	return response.json()
 
 
-def run_model(text):
+def run_model(Dialects):
 	global hold_time
 	output = query({
-	"inputs": text,
+	"inputs": Dialects,
 	"parameters": {}
 })
 	if output:
 		hold_time = 1
 		return output[0]['generated_text']
 	else:
-		return f'Model is being loaded, please try again in {int((hold_time - time.time()) + 35)} seconds.'
+		wait_time = int((hold_time - time.time()) + 35)
+		if wait_time <= 0:
+			return f'Model is being loaded, please try again in {wait_time} seconds.'
+		else:
+			return 'Taking longer than usual to load please wait or refresh the page.'
 		
 	
 
 
-run_model('السلام عيكم')
-
-time.sleep(35)
-
-
+# run_model('السلام عيكم')
 
+# time.sleep(35)
 
 
-import gradio as gr
-
-examples = [
+examples_text = [
     ["ما ابغا أروح الإمتحان"],
     ["أييد أن انام ف لبيتنا"],
-	["Hello how are you today"]
+	["Hello how are you today"],
+	["kef al7al"]
 ]
 
+
 def mode_run(text):
 	result = run_model(text)
 	return result
 
+link = '<a href="{}" target="_blank" style="cursor: pointer; font-size: 18px;">{}</a>'
+
+with gr.Blocks(theme=gr.themes.GoogleFont('ali')) as demo:
+
+
+
+	gr.Markdown(
+		"""
+		## Dialects to MSA transformer
+		Start typing Non-Traditional Arabic to convert into Classical version.
+		"""
+	)
+
+	with gr.Row():
+		with gr.Column():
+			input = gr.Textbox(label='Dialects')
+		with gr.Column():
+			output = gr.Textbox(label='MSA')
+		
+		
+
+	with gr.Row():
+		button = gr.Button('Submit',variant='primary')
+		clear = gr.ClearButton(input)
+		
+
+	examples = gr.Examples(examples_text,input)
+	
+	
+	with gr.Row():
+		gr.Markdown(
+			"""
+			## Model Overview
+			This Model is optimized to convert written text in various non Standard Classical Arabic into Classic Arabic, the model was Fine-Tuned on 0.8M pairs of sentence generated by OpenAI API gpt-4o-mini Text Generation Model, beside being able to convert Dialects into Classical Arabic, the model can also be used in other NLP tasks such as Text Correction, Diacretization and Sentence Punctuation.
+			"""
+		)
+
+	
+
 
-demo = gr.Interface(fn=mode_run,
-					inputs="text",
-					outputs="text",
-					examples=examples)
+	with gr.Row():
+		gr.Markdown(
+			"""
+			## Dielcts the Model trained on
+			Below image shows an estimate of dialects the model trained on.
+			"""
+		)
+	with gr.Row():
+		with gr.Column(scale=3):
+			gr.Image(Image.open('Dialects by Region.png'),height=300,container=False)
+		with gr.Column(scale=3):
+			pass
+
+
+
+	with gr.Row():
+		with gr.Column(scale=1):
+			pass
+		with gr.Column(scale=2):
+			gr.HTML(
+				'<div style="text-align: center;">' +\
+				link.format('https://huggingface.co/HamzaNaser/Dialects-to-MSA-Transformer', 'Model Card') + ' -- '+link.format('https://www.linkedin.com/in/hamza-naser-0b4b90160/', 'LinkedIn') +\
+				'</div>'
+				)
+		with gr.Column(scale=1):
+			pass
+
+
+	input.submit(mode_run,input,output)
+	button.click(mode_run,input,output)
 
 demo.launch()