Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 hf_token = os.getenv("HF_Token")
@@ -17,11 +18,11 @@ def Sentence_Commpletion(model_name, input):
     if model_name == "Bloom":
         tokenizer, model = b_tokenizer, b_model
         inputs = tokenizer(input, return_tensors="pt")
-        outputs = model.generate(inputs.input_id, max_length=30, num_return_sequences=1)
+        outputs = model.generate(inputs.input_ids, max_length=30, num_return_sequences=1)
     elif model_name == "Gemma":
         tokenizer, model = g_tokenizer, g_model
         inputs = Tokenizer(input, return_tensors="pt")
-        outputs = Model.generate(**input_id, max_new_tokens=20)
+        outputs = Model.generate(**input_ids, max_new_tokens=20)
     return tokenizer.decode(outputs[0])