update

app.py

@@ -152,9 +152,6 @@ class CustomModel(nn.Module):
 
         return output.logits
     
-model = CustomModel()
-
-
 
 ### End Load the model
 
@@ -171,6 +168,7 @@ def greet():
     pass
 
 with gr.Blocks(title=title) as demo: # theme=theme
+    model = CustomModel()
     sample_examples = pd.read_csv('sample_examples.csv')
     example_list = sample_examples[['Question','Response','target']].sample(2).values.tolist()
     gr.Markdown(f"## {title}")