Update app.py

app.py

@@ -1,6 +1,18 @@
 import gradio as gr
 from tridentmodel import classification
 import pandas as pd
+import torch
+import torch.nn as nn
+import transformers
+from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
+
+tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara")
+
+model = LlamaForCausalLM.from_pretrained(
+    "Claimed/capybara")
+#model = model.to('cuda')
+
+
 
 def add_text(history, text):
     history = history + [(text, None)]