Commenting out the LLM part

app.py

@@ -9,29 +9,29 @@ import transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-
-model = "tiiuae/falcon-7b-instruct"
-
-tokenizer = AutoTokenizer.from_pretrained(model)
-pipeline = transformers.pipeline(
-    "text-generation",
-    model=model,
-    tokenizer=tokenizer,
-    torch_dtype=torch.bfloat16,
-    trust_remote_code=True,
-    device_map="auto",
-)
-sequences = pipeline(
-   "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
-    max_length=200,
-    do_sample=True,
-    top_k=10,
-    num_return_sequences=1,
-    eos_token_id=tokenizer.eos_token_id,
-)
-st.title("Beyond the Anti-Jam: Integration of DRL with LLM")
-for seq in sequences:
-    st.write(f"Result: {seq['generated_text']}")
+#
+# model = "tiiuae/falcon-7b-instruct"
+#
+# tokenizer = AutoTokenizer.from_pretrained(model)
+# pipeline = transformers.pipeline(
+#     "text-generation",
+#     model=model,
+#     tokenizer=tokenizer,
+#     torch_dtype=torch.bfloat16,
+#     trust_remote_code=True,
+#     device_map="auto",
+# )
+# sequences = pipeline(
+#    "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
+#     max_length=200,
+#     do_sample=True,
+#     top_k=10,
+#     num_return_sequences=1,
+#     eos_token_id=tokenizer.eos_token_id,
+# )
+# st.title("Beyond the Anti-Jam: Integration of DRL with LLM")
+# for seq in sequences:
+#     st.write(f"Result: {seq['generated_text']}")
 
 
 def perform_training(jammer_type, channel_switching_cost):

requirements.txt

@@ -3,7 +3,7 @@ tensorflow
 matplotlib
 gym
 streamlit
-transformers==4.6.0
+transformers
 torch
 einops
 accelerate