Update generate_model.py

generate_model.py

@@ -706,11 +706,9 @@ def tinyllava_elm_generate_parser():
 
 if __name__ == '__main__':
     args = tinyllava_elm_generate_parser()
-    prompt = args.prompt
-    model = TinyLlavaForConditionalGeneration.from_pretrained(args.model)
 
     output_text, genertaion_time = generate(
-        prompt=prompt,
+        prompt=args.prompt,
         image=args.image,
         model=args.model,
         device=args.device,