Update app.py

app.py

@@ -3,12 +3,16 @@ import gradio as gr
 import os
 import transformers
 from transformers import AutoModel, AutoTokenizer
+import numpy as np
 
 model_names = ['python']
 
 models = {}
 model = AutoModel.from_pretrained(f'ZarahShibli/tmp_trainer')
 
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+MAX_LEN = 200
+
 tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', return_tensors='pt')
 
 target_cols = [ 'DevelopmentNotes', 'Expand','Parameters', 'Summary', 'Usage']