Update app.py

app.py

@@ -1,8 +1,13 @@
 # app.py
-from flask import Flask, render_template, request, jsonify
+#from flask import Flask, render_template, request, jsonify
+
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
 from transformers import pipeline
 
-app = Flask(__name__)
+app = FastAPI()
 #model = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad', revision='main')
 
 @app.route('/')
@@ -16,5 +21,5 @@ def predict():
     #response_text = result[0]['generated_text']
     return jsonify({'response': "Hello, Hugging Face!"})#response_text
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+#if __name__ == '__main__':
+#    app.run(host='0.0.0.0', port=5000)