Update app.py

app.py

@@ -83,19 +83,18 @@ import time
 
 def get_prompts(prompt_text):
     if not prompt_text:
-        return "Please enter text before generating prompts.رجاء ادخل النص اولا" 
-        raise gr.Error("Please enter text before generating prompts.رجاء ادخل النص اولا")
+        return "Please enter text before generating prompts.رجاء ادخل النص اولا"
     else:
         global request_counter
         request_counter += 1
         timestamp = f"{time.time()}_{request_counter}"
 
-        options = {"src": "ar", "tgt": "en"}
-        options.update({"timestamp": timestamp})
-
-        # Adjust this line if needed based on the model's input/output specifications
-        return text_gen(prompt_text, options=options)
-
+        tokenizer.src_lang = "ar_AR"
+        encoded_ar = tokenizer(prompt_text, return_tensors="pt")
+        generated_tokens = model.generate(**encoded_ar)
+        translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+        
+        return translated_text
 
 
 # Existing code...