Spaces:
Runtime error
Runtime error
AkshayaKeerthi
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from flask import Flask, render_template, request, jsonify
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
app = Flask(__name__)
|
6 |
-
model = pipeline('conversational', model='
|
7 |
|
8 |
@app.route('/')
|
9 |
def home():
|
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
app = Flask(__name__)
|
6 |
+
model = pipeline('conversational', model='microsoft/DialoGPT-small', revision='main')
|
7 |
|
8 |
@app.route('/')
|
9 |
def home():
|