Update main.py

main.py

@@ -1,4 +1,4 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+
 from flask import Flask, request, jsonify
 
 device = "cuda" # the device to load the model onto