Update app.py

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 
-import torch
+from inductor import BartInductor
 
-if torch.cuda.is_available():
-    device = torch.device("cuda")
-    print('GPU is available')
-else:
-    device = torch.device("cpu")
-    print('GPU is not available')
+inductor = BartInductor()
+
+rule = '<mask> is the capital of <mask>.'
+generated_texts = inductor.generate(rule)
+
+for text in generated_texts:
+    print(text)
 
 def uppercase(text):
     return text.upper()