Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import numpy as np
+import gradio as gr
 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
@@ -120,6 +121,7 @@ model = AutoModelForCausalLM.from_pretrained("gpt2")
 
 tokenizer.pad_token_id = tokenizer.eos_token_id
 print("Loading finished.")
+
 def generate_html(token, node):
     """Recursively generate HTML for the tree."""
 
@@ -220,8 +222,6 @@ def get_tables(input_text, number_steps, number_beams):
     )
     return tables
     
-import gradio as gr
-
 with gr.Blocks(
     theme=gr.themes.Soft(
         text_size="lg", font=["monospace"], primary_hue=gr.themes.colors.green