remove cuda

print debug language

image.py

@@ -15,10 +15,9 @@ def image(url, text):
     outputs = model(**encoding)
     logits = outputs.logits
     idx = logits.argmax(-1).item()
-    # print("question asked:", text)
-    # print("image link:", url)
-    # print("Predicted answer:", model.config.id2label[idx])
-
+    print("question asked:", text)
+    print("image link:", url)
+    print("Predicted answer:", model.config.id2label[idx])
     return model.config.id2label[idx]
 
 # prepare image + question

language.py

@@ -6,8 +6,8 @@ def longText(answere, question):
     input_text = "i have a question and answer.\nthe question is : {}\n the response is : {}\n with this information, can you create an answer phrase?".format(question, answere)
 
     tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
-    model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large", device_map="auto")
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
+    model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
     outputs = model.generate(input_ids)
 
     return tokenizer.decode(outputs[0]).replace("<pad>","").replace("</s>","")