Update app.py

app.py

@@ -17,7 +17,7 @@ titik10=gr.Number(label='Node 10 (Close)')
 titik11=gr.Number(label='Node 11 (Close)')
 desc='For one-time prediction only. EURUSD TF 1D for now.'
 net=joblib.load('model_savgolfilt')
-def fn(titik1,titik2,titik3,titik4,titik5,titik6,titik7,titik8,titik9,titik10,titik11):
+def fn(pair,titik1,titik2,titik3,titik4,titik5,titik6,titik7,titik8,titik9,titik10,titik11):
     titik=np.array([])
     titik=np.append(titik,titik1)
     titik=np.append(titik,titik2)
@@ -31,8 +31,13 @@ def fn(titik1,titik2,titik3,titik4,titik5,titik6,titik7,titik8,titik9,titik10,ti
     titik=np.append(titik,titik10)
     titik=np.append(titik,titik11)
     x=savgol_filter(titik,11,3)
-    hasil=net.predict([[x[len(x)-2],x[len(x)-1]]])
-    keluar=px.line(y=[x[len(x)-2],x[len(x)-1],hasil[0]])
+    if pair="With Last Node":
+        hasil=net.predict([[x[len(x)-2],x[len(x)-1]]])
+        keluar=px.line(y=[x[len(x)-2],x[len(x)-1],hasil[0]])
+    elif pair="Without Last Node":
+        hasil=net.predict([[x[len(x)-3],x[len(x)-2]]])
+        keluar=px.line(y=[x[len(x)-3],x[len(x)-2],hasil[0]])
     return keluar
-app=gr.Interface(fn,inputs=[titik1,titik2,titik3,titik4,titik5,titik6,titik7,titik8,titik9,titik10,titik11],outputs=[gr.Plot()],description=desc,title='Savitsky-Golay Prediction')
+pair=["With Last Node","Without Last Node"]
+app=gr.Interface(fn,inputs=[pair,titik1,titik2,titik3,titik4,titik5,titik6,titik7,titik8,titik9,titik10,titik11],outputs=[gr.Plot()],description=desc,title='Savitsky-Golay Prediction')
 app.launch()