update app.py

app.py

@@ -1,3 +1,27 @@
+
+import torch
+import RohanGivenCode
+from RohanGivenCode import *
+
+
+save1_or_load0 = 0 # 1 => Save; 0 => Load
+
+device = 'cpu'
+if torch.cuda.is_available():
+    device = 'cuda'
+elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
+    device = "mps"
+print(f"using device: {device}")
+
+# SEED
+torch.manual_seed(1337)
+if torch.cuda.is_available():
+    torch.cuda.manual_seed(1337)
+
+# STOP
+# num_return_sequences = 5
+# max_length = 30
+
 import gradio as gr
 def sentence_builder(new_tokens, ret_seq, txt):
     txt_len = len(txt.split())