Update app.py

app.py

@@ -62,13 +62,11 @@ def translate(
     
     prompt = Prompt_template(source_text, source_lang, target_lang)
     inputs = tokenizer(prompt, return_tensors="pt")
-    
-    input_ids = inputs.to(model.device)
-    
+        
     streamer = TextIteratorStreamer(tokenizer, **{"skip_special_tokens": True, "skip_prompt": True, 'clean_up_tokenization_spaces':False,})
 
     generate_kwargs = dict(
-        input_ids=input_ids, 
+        input_ids=inputs, 
         streamer=streamer,
         max_length=max_length, 
         do_sample=True, 
@@ -134,8 +132,8 @@ with gr.Blocks(theme="soft", css=CSS) as demo:
                 value=0.3,
                 step=0.1,
             )
-        with gr.Column(scale=4):
             gr.Markdown(DESCRIPTION)
+        with gr.Column(scale=4):
             source_text = gr.Textbox(
                 label="Source Text",
                 value="How we live is so different from how we ought to live that he who studies "+\